diff --git a/pyproject.toml b/pyproject.toml
index f796468..c38d13b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -106,6 +106,9 @@ cov = [
 docs = "mkdocs serve -a localhost:8001 {args}"
 
 # TOOLS
+[tool.hatch.build.targets.wheel]
+packages = ["src/trapi_predict_kit"]
+
 [tool.hatch.version]
 path = "src/trapi_predict_kit/__init__.py"