diff --git a/pyproject.toml b/pyproject.toml
index 0f2748d..56c22e7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,4 +1,7 @@
 [tool]
+[tool.setuptools]
+packages = ['ml_experiment']
+
 [tool.commitizen]
 name = "cz_conventional_commits"
 version = "0.0.0"