diff --git a/pyproject.toml b/pyproject.toml
index e30e8f7..d78b677 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -82,7 +82,10 @@ include-package-data = true
 zip-safe = false
 
 [tool.setuptools.packages.find]
-include = ["vessel_analysis_3d"]
+include = ["vessel_analysis_3d", "vessel_analysis_3d.*"]
+
+[tool.setuptools.package-dir]
+vessel_analysis_3d = "vessel_analysis_3d"
 
 [project.scripts]
 run_vessel_analysis = "vessel_analysis_3d.bin.run_analysis:main"
@@ -115,4 +118,4 @@ ignore = [
 	"W292",
 	"E501",
 ] 
-max-line-length = 88
\ No newline at end of file
+max-line-length = 88