diff --git a/pyproject.toml b/pyproject.toml
index 3686532ed..5a1d7a1d1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -107,6 +107,9 @@ xarm = ["gym-xarm>=0.1.1 ; python_version < '4.0'"]
 
 [tool.poetry]
 requires-poetry = ">=2.1"
+packages = [
+    { include = "lerobot", from = "src" }
+]
 
 [tool.ruff]
 line-length = 110