diff --git a/aidge_core/export_utils/node_export.py b/aidge_core/export_utils/node_export.py
index eeba09c92d6904b49435b71694cc0261976e7879..1154e88f393a0312d9b82d763c368de047560eab 100644
--- a/aidge_core/export_utils/node_export.py
+++ b/aidge_core/export_utils/node_export.py
@@ -240,11 +240,11 @@ class ExportNode(ABC):
                     if "cont_size" in mem_info[idx]:
                         self.attributes["mem_info_cont_size"][idx] = mem_info[idx]["cont_size"]
                     else:
-                        self.attributes["mem_info_cont_size"][idx] = tensor.size()
+                        self.attributes["mem_info_cont_size"][idx] = mem_info[idx]["size"]
                     if "cont_offset" in mem_info[idx]:
                         self.attributes["mem_info_cont_offset"][idx] = mem_info[idx]["cont_offset"]
                     else:
-                        self.attributes["mem_info_cont_offset"][idx] = 0
+                        self.attributes["mem_info_cont_offset"][idx] = mem_info[idx]["offset"]
                     if "cont_offset" in mem_info[idx]:
                         self.attributes["mem_info_wrap_offset"][idx] = mem_info[idx]["wrap_offset"]
                     else: