diff --git a/pyproject.toml b/pyproject.toml
index 0d4e4d6..fff020c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,7 +13,7 @@ include = ["terratorch_iterate*"]
 [project]
 
 name = "terratorch-iterate"
-version = "0.2.2rc1"
+version = "0.2.3"
 requires-python = ">= 3.11"
 description = "A terratorch's plugin for benchmarking and hyperparameter optimization"
 authors = [
@@ -64,7 +64,7 @@ dependencies = [
 "mlflow",
 "optuna",
 "types-tabulate",
-"ray",
+"ray<=2.49.2",
 "gputil",
 "asyncio",
 "opencv-python-headless",
@@ -72,6 +72,8 @@ dependencies = [
 "optuna-integration",
 "psutil",
 "tabulate>=0.9.0",
+"torch",
+"seaborn"
 ]
 
 [project.urls]
@@ -87,8 +89,7 @@ dev = [
   "mike", # for building docs with versions
   "tox",
   "pre-commit",
-  "seaborn"
-]
+  ]
 
 test = [
   "coverage",
@@ -112,9 +113,6 @@ skip-string-normalization = true
 
 [project.scripts]
 iterate = "terratorch_iterate.main:main"
-# ray_benchmark = "benchmark.benchmark_ray:main"
-# repeat_experiments = "benchmark.main:main"
-
 
 [tool.isort]
 multi_line_output = 3
diff --git a/terratorch_iterate/main.py b/terratorch_iterate/main.py
index 8b23c4b..0d411c1 100644
--- a/terratorch_iterate/main.py
+++ b/terratorch_iterate/main.py
@@ -246,7 +246,7 @@ def main():
         if config_path is None:
             msg = """
             Error: config argument has not been passed
-            usage: terratorch iterate [-h] [--hpo] [--repeat] [--summarize] [--config CONFIG] 
+            Usage: iterate [-h] [--hpo] [--repeat] [--summarize] [--config CONFIG] 
             """
             print(msg)
         else: