diff --git a/aidge_export_arm_cortexm/export.py b/aidge_export_arm_cortexm/export.py
index 979e19458860d05299a54d9591e8a9fc62885812..110e9431924ae6ad45a9d6fa852a1201e4779428 100644
--- a/aidge_export_arm_cortexm/export.py
+++ b/aidge_export_arm_cortexm/export.py
@@ -3,9 +3,11 @@ import shutil
 from pathlib import Path
 from aidge_export_arm_cortexm.utils import (ROOT, AVAILABLE_BOARDS, has_board)
 from aidge_export_arm_cortexm.export_registry import ExportLibAidgeARM
-from aidge_core.mem_info import compute_default_mem_info
+# from aidge_export_arm_cortexm.utils.converter import numpy_dtype2ctype
+
+from aidge_core.mem_info import compute_default_mem_info, generate_optimized_memory_info
 from aidge_core.export_utils import scheduler_export
-# from aidge_core.export_utils.code_generation import *
+
 
 def export(export_folder_name,
            graphview,
@@ -18,7 +20,8 @@ def export(export_folder_name,
         scheduler,
         export_folder_name,
         ExportLibAidgeARM,
-        memory_manager=compute_default_mem_info
+        memory_manager=generate_optimized_memory_info,
+        memory_manager_args={"stats_folder": f"{export_folder_name}/stats", "wrapping":False }
     )
     # Create export directory
     export_folder = Path().absolute() / export_folder_name