diff --git a/pyproject.toml b/pyproject.toml
index 06f0490..4680776 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -17,6 +17,8 @@ numpy = "^1.26.4"
 scipy = "^1.13.0"
 matplotlib = "^3.9.0"
 tensorboard = "^2.16.2"
+click = "^8.1.7"
+pyyaml = "^6.0.1"
 
 [build-system]
 requires = ["poetry-core"]
@@ -55,7 +57,7 @@ m2m = "map2map:main.main"
 mapcropper = "map2map:cropper.cropper"
 
 [tool.poetry.scripts]
-map2map = "map2map:main"
+map2map = "map2map:main.main"
 
 
 [project.urls]