diff --git a/setup.py b/setup.py
index af99bcfe10540cb26bc1a6f029cd0cda2e5035e2..ff3ba72b2eab8a796638e2ffb7db3059297549da 100644
--- a/setup.py
+++ b/setup.py
@@ -93,8 +93,8 @@ setuptools.setup(
         "matplotlib",
         "scipy"
     ],
-    packages=["b_asic", "b_asic/GUI"],
-    ext_modules=[CMakeExtension("b_asic")],
+    packages=["b_asic", "b_asic/GUI", "b_asic/scheduler_gui"],
+    ext_modules=[CMakeExtension("_b_asic")],
     cmdclass={"build_ext": CMakeBuild},
     zip_safe=False,
     include_package_data=True