diff --git a/setup.py b/setup.py
index 6f47af695545868600705503f1a093d12d54ea9d..0284d1fd037c4e6f7c525df5b29c1872d384e6d7 100644
--- a/setup.py
+++ b/setup.py
@@ -6,14 +6,14 @@ with open("README.md", "r", encoding="utf-8") as f:
 
 setup(
     name="pt2d",
-    version="0.1.0",
+    version="0.1.1",
     author="Aske T. Rove, Christian L. Bjerregaard, Mikkel W. Breinstrup",
     author_email="s224362@dtu.dk, s224389@dtu.dk, s224361@dtu.dk",
     description="Interactive path tracing in 2D medical images",
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://lab.compute.dtu.dk/QIM/tools/pt2d",
-    packages=find_packages(),
+    packages=find_packages(include=["pt2d", "pt2d.*"]),
     include_package_data=True,
     entry_points = {
         'console_scripts': [