diff --git a/pyproject.toml b/pyproject.toml
index 13f7a6c..6cb3035 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -54,6 +54,9 @@ dependencies = [
 [project.scripts]
     pyvisgen_create_dataset = "pyvisgen.simulation.scripts.create_dataset:main"
 
+[tool.setuptools.packages.find]
+    where = ["pyvisgen"]
+
 [tool.towncrier]
     package = "ctapipe"
     directory = "docs/changes"