diff --git a/pyproject.toml b/pyproject.toml
index 594ea08..90006bd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,8 +29,9 @@ dependencies = [
 requires = ["setuptools"]
 build-backend = "setuptools.build_meta"
 
-[tool.setuptools]
-packages = ["orb_models"]
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["orb_models*"]
 
 [project.urls]
 Homepage = "https://github.com/orbital-materials/orb-models"