diff --git a/setup.py b/setup.py
index f029f847055e9683457e11e8e0bdd22017310cb2..03d9afcb65fc9da6e84231cb11b252cfab58add5 100644
--- a/setup.py
+++ b/setup.py
@@ -27,8 +27,7 @@ setup(name="nifty2go",
       author_email="theos@mpa-garching.mpg.de",
       description="Numerical Information Field Theory",
       url="http://www.mpa-garching.mpg.de/ift/nifty/",
-      packages=find_packages(),
-      package_dir={"nifty2go": "nifty2go"},
+      packages=find_packages(include=["nifty2go", "nifty2go.*"]),
       zip_safe=False,
       license="GPLv3",
       classifiers=[