diff --git a/.gitignore b/.gitignore
index 9fbfccca6dfda997d8a0dbfc4b373590feeecad8..f37378e300efeb5362882eb8d6eb59f028563a0e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -10,6 +10,7 @@ install*/
 __pycache__
 *.pyc
 *.egg-info
+dist*/
 
 # Mermaid
 *.mmd
diff --git a/aidge_core/__init__.py b/aidge_core/__init__.py
index 3b9dbbf8f1ecfd3cfd0cbe05001b98316cd9e8d4..8e0bef3a02be05199c0092461b920fe7e5d839dd 100644
--- a/aidge_core/__init__.py
+++ b/aidge_core/__init__.py
@@ -1 +1 @@
-from .aidge_core import * # import so generated by PyBind
\ No newline at end of file
+from aidge_core.aidge_core import * # import so generated by PyBind
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 7dbde6a0ae8d2b4595f08614d964b0d382030c94..cc3d5dbb810c74f5d8c88bd9258c139fa9a30861 100644
--- a/setup.py
+++ b/setup.py
@@ -102,10 +102,10 @@ if __name__ == '__main__':
         platforms=["Linux"],
         packages=find_packages(where="."),
         include_package_data=True,
-        # exclude_package_data={"aidge": ["_Core/*", "_CPU/*"]},
-        ext_modules=[CMakeExtension('aidge_core')],
+        ext_modules=[CMakeExtension(get_project_name())],
         cmdclass={
             'build_ext': CMakeBuild,
         },
+        zip_safe=False,
 
     )