diff --git a/pyproject.toml b/pyproject.toml
index 5b71ef93..a70074ce 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -59,6 +59,8 @@ path = "src/__about__.py"
 packages = [
 	"src/server", 
 	"src/estimate",
+	"src/util",
+	"src/train",
 ]
 
 [tool.hatch.envs.default]