diff --git a/projective_simulation/__init__.py b/projective_simulation/__init__.py
index 3dc1f76..485f44a 100644
--- a/projective_simulation/__init__.py
+++ b/projective_simulation/__init__.py
@@ -1 +1 @@
-__version__ = "0.1.0"
+__version__ = "0.1.1"
diff --git a/pyproject.toml b/pyproject.toml
index 8e9dd89..97a8958 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,7 +25,7 @@ projective_simulation = "projective_simulation._modidx:d"
 version = {attr = "projective_simulation.__version__"}
 
 [tool.setuptools.packages.find]
-include = ["projective_simulation"]
+include = ["projective_simulation", "projective_simulation.*"]
 
 [tool.nbdev]
 doc_path = 'docs'