diff --git a/aidge_core/unit_tests/test_export.py b/aidge_core/unit_tests/test_export.py
index 23a65c8a2e0204d1f8a60b97673a3ba26ef131b3..32d902b5feedebb3c4e27b6fae0da07be77175fa 100644
--- a/aidge_core/unit_tests/test_export.py
+++ b/aidge_core/unit_tests/test_export.py
@@ -88,6 +88,7 @@ class test_export(unittest.TestCase):
         )
 
         initFiller(model)
+        model.forward_dims([[1, 32*32*3]])
 
         # Export model
         aidge_core.serialize_to_cpp(self.EXPORT_PATH, model)