diff --git a/benchmark/benchmark/start_benchmark.py b/benchmark/benchmark/start_benchmark.py
index 4235e67a..b7cda4bd 100644
--- a/benchmark/benchmark/start_benchmark.py
+++ b/benchmark/benchmark/start_benchmark.py
@@ -430,6 +430,3 @@ def version():
 #         latest_report = json.load(file)
 
 #     return latest_report
-
-if __name__ == "__main__":
-    cli()
diff --git a/benchmark/pyproject.toml b/benchmark/pyproject.toml
index 7fd2efcf..caeee5fc 100644
--- a/benchmark/pyproject.toml
+++ b/benchmark/pyproject.toml
@@ -88,3 +88,6 @@ ensure_newline_before_comments = true
 line_length = 88
 sections = ["FUTURE", "STDLIB", "THIRDPARTY", "FIRSTPARTY", "LOCALFOLDER"]
 skip_glob = [".tox", "__pycache__", "*.pyc", "venv*/*", "reports", "venv", "env", "node_modules", ".env", ".venv", "dist", "agent/*", "agbenchmark/challenges/*"]
+
+[tool.poetry.scripts]
+agbenchmark = "benchmark.start_benchmark:cli"