diff --git a/aidge_export_cpp/export.py b/aidge_export_cpp/export.py
index a805fb22e56048a718892344f6ae5cd43bbf5ff7..4cd93b85466d8008c8d44d0b57ee428a5f1e0c5a 100644
--- a/aidge_export_cpp/export.py
+++ b/aidge_export_cpp/export.py
@@ -79,6 +79,7 @@ def export(export_folder_name: str,
     if inputs_tensor is not None:
         if inputs_tensor.dtype() == aidge_core.dtype.int32:
             set_nodes_datatypes(model)      # Set datatype to int8 only
+            inputs_tensor.set_datatype(aidge_core.dtype.int8)
             model.save(f"graph_log/3_set_datatypes")
 
     # [aidge_cmp] Export feature maps tensors as json