diff --git a/pyproject.toml b/pyproject.toml
index e7163e4..1a0d2df 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,9 @@
 [tool.poetry]
 name = "timesfm"
-packages = [{ include = "timesfm", from = "src" }]
+packages = [
+    { include = "timesfm", from = "src" },
+    { include = "finetuning", from = "src" },
+]
 description = "Open weights time-series foundation model from Google Research."
 version = "1.2.8"
 authors = [