diff --git a/setup.py b/setup.py
index 8082341..68173b6 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,7 @@
 import io
 import os
 import sys
-from setuptools import setup, find_packages
+from setuptools import setup
 
 version_info = {}
 with open(os.path.join("src", "relaxit", "_version.py")) as f:
@@ -24,7 +24,13 @@
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/intsystems/discrete-variables-relaxation",
-    package_dir= {"" : "src"} , 
-    packages=find_packages(where="src"),
+    packages=[
+        "relaxit", 
+        "relaxit.distributions"
+    ],
+    package_dir={
+        "relaxit": "src/relaxit",
+        "relaxit.distributions": "src/relaxit/distributions"
+    }, 
     install_requires=["pyro-ppl==1.9.1"],
 )
\ No newline at end of file