diff --git a/aidge_export_arm_cortexm/export.py b/aidge_export_arm_cortexm/export.py
index f211fbbe957f55629bc48b77999b12c7c48d2c8a..92ad2385a3f9df16b89462c58bd8b06f15520d3f 100644
--- a/aidge_export_arm_cortexm/export.py
+++ b/aidge_export_arm_cortexm/export.py
@@ -4,7 +4,7 @@ import shutil
 from pathlib import Path
 import numpy as np
 
-from aidge_core.export.code_generation import *
+from aidge_core.export_utils.code_generation import *
 from aidge_export_arm_cortexm.utils import (ROOT, AVAILABLE_BOARDS, has_board, \
                                             OPERATORS_REGISTRY, supported_operators)
 import aidge_export_arm_cortexm.operators
diff --git a/aidge_export_arm_cortexm/operators.py b/aidge_export_arm_cortexm/operators.py
index 0d2a3c6f7f5773684ea71a8d562d97243cd5fae5..f9520fefa11d6fdb6c7300cae7e5530cb5af0032 100644
--- a/aidge_export_arm_cortexm/operators.py
+++ b/aidge_export_arm_cortexm/operators.py
@@ -8,7 +8,7 @@ from typing import Tuple, List, Union, Dict
 
 import aidge_core
 from aidge_core import ExportNode
-from aidge_core.export.code_generation import *
+from aidge_core.export_utils.code_generation import *
 from aidge_export_arm_cortexm.utils import ROOT, operator_register
 from aidge_export_arm_cortexm.utils.converter import numpy_dtype2ctype, aidge_datatype2dataformat, aidge_datatype2ctype
 from aidge_export_arm_cortexm.utils.generation import *