diff --git a/aidge_core/unit_tests/test_export.py b/aidge_core/unit_tests/test_export.py
index febeef96c0158f242cce112f723c391d623f7e4d..a0929eb96e02f65855eb60891899b8403cf43d2e 100644
--- a/aidge_core/unit_tests/test_export.py
+++ b/aidge_core/unit_tests/test_export.py
@@ -10,14 +10,11 @@ SPDX-License-Identifier: EPL-2.0
 
 import unittest
 import aidge_core
-from functools import reduce
 import pathlib
 import os
 import sys
 import subprocess
 import shutil
-import numpy as np
-
 
 def initFiller(model):
     # Initialize parameters (weights and biases)
@@ -43,6 +40,17 @@ def initFiller(model):
             else:
                 pass
 
+def clean_dir(dir : pathlib.Path) -> None:
+    for filename in os.listdir(dir):
+        file_path = os.path.join(dir, filename)
+        try:
+            if os.path.isfile(file_path) or os.path.islink(file_path):
+                os.unlink(file_path)
+            elif os.path.isdir(file_path):
+                shutil.rmtree(file_path)
+        except Exception as e:
+            print('Failed to delete %s. Reason: %s' % (file_path, e))
+    return
 
 class test_export(unittest.TestCase):
     """Test aidge export"""
@@ -74,6 +82,9 @@ class test_export(unittest.TestCase):
 
         # Export model
         aidge_core.export(self.EXPORT_PATH, model)
+
+        
+        clean_dir(self.EXPORT_PATH / "build")
         self.assertTrue(
             self.EXPORT_PATH.is_dir(), "Export folder has not been generated"
         )