diff --git a/aidge_core/export_utils/data_conversion.py b/aidge_core/export_utils/data_conversion.py
index 2b1b0664b50ff17e1db880bd9c6bc680160d3a66..401fc39f2a70245a67719699b5f0cdc61108e0cf 100644
--- a/aidge_core/export_utils/data_conversion.py
+++ b/aidge_core/export_utils/data_conversion.py
@@ -3,17 +3,17 @@ import aidge_core
 
 
 datatype_converter_aide2c = {
-    aidge_core.DataType.Float64 : "double",
-    aidge_core.DataType.Float32 : "float",
-    aidge_core.DataType.Float16 : "half_float::half",
-    aidge_core.DataType.Int8    : "int8_t",
-    aidge_core.DataType.Int16   : "int16_t",
-    aidge_core.DataType.Int32   : "int32_t",
-    aidge_core.DataType.Int64   : "int64_t",
-    aidge_core.DataType.UInt8   : "uint8_t",
-    aidge_core.DataType.UInt16  : "uint16_t",
-    aidge_core.DataType.UInt32  : "uint32_t",
-    aidge_core.DataType.UInt64  : "uint64_t"
+    aidge_core.dtype.float64 : "double",
+    aidge_core.dtype.float32 : "float",
+    aidge_core.dtype.float16 : "half_float::half",
+    aidge_core.dtype.int8    : "int8_t",
+    aidge_core.dtype.int16   : "int16_t",
+    aidge_core.dtype.int32   : "int32_t",
+    aidge_core.dtype.int64   : "int64_t",
+    aidge_core.dtype.uint8   : "uint8_t",
+    aidge_core.dtype.uint16  : "uint16_t",
+    aidge_core.dtype.uint32  : "uint32_t",
+    aidge_core.dtype.uint64  : "uint64_t"
 }
 
 def aidge2c(datatype):