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