diff --git a/setup.py b/setup.py
index 5579b6a..8082341 100644
--- a/setup.py
+++ b/setup.py
@@ -1,16 +1,12 @@
 import io
 import os
 import sys
-from setuptools import setup
+from setuptools import setup, find_packages
 
 version_info = {}
 with open(os.path.join("src", "relaxit", "_version.py")) as f:
     exec(f.read(), version_info)
 
-def read(file_path):
-    with io.open(file_path, "r", encoding="utf-8") as f:
-        return f.read()
-
 try:
     long_description = open("README.md", encoding="utf-8").read()
 except Exception as e:
@@ -28,7 +24,7 @@ def read(file_path):
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/intsystems/discrete-variables-relaxation",
-    packages=["relaxit"],
-    package_dir={"relaxit": "src/relaxit"}, 
+    package_dir= {"" : "src"} , 
+    packages=find_packages(where="src"),
     install_requires=["pyro-ppl==1.9.1"],
 )
\ No newline at end of file