diff --git a/pyproject.toml b/pyproject.toml
index a12e635..5af18f0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,5 +1,8 @@
 [tool.poetry]
 name = "timesfm"
+packages = [
+    { include = "timesfm", from = "src" },
+]
 version = "0.0.2"
 description = "Open weights time-series foundation model from Google Research."
 authors = [