diff --git a/setup.py b/setup.py
index 79e8825..57c0c98 100644
--- a/setup.py
+++ b/setup.py
@@ -7,6 +7,7 @@ setup(
     author='Dominik Roth',
     # author_email='author@gmail.com',
     # description='Description of my package',
-    packages=['priorConditionedAnnealing'],
+    #packages=['priorConditionedAnnealing'],
+    packages=find_packages(),
     install_requires=['numpy', 'scipy', 'torch', 'perlin_noise', 'colorednoise', 'stable_baselines3'],
 )