gistfile0.txt

diff --git a/testsuite/replay-verify/main.py b/testsuite/replay-verify/main.py
index ea7b157676..26564d067b 100644
--- a/testsuite/replay-verify/main.py
+++ b/testsuite/replay-verify/main.py
@@ -290,9 +290,13 @@ class ReplayScheduler:
         return (self.failed_workpod_logs, self.txn_mismatch_logs)
 
 
-if __name__ == "__main__":
+def main():
     scheduler = ReplayScheduler(
         "test", 1000000000, 1000500000, [(1000, 2000), (3000, 4000)], range_size=200_000
     )
     scheduler.schedule(from_scratch=True)
     print(scheduler.collect_all_failed_logs())
+
+
+if __name__ == "__main__":
+    main()
diff --git a/testsuite/replay-verify/pyproject.toml b/testsuite/replay-verify/pyproject.toml
index 39cdaf8b91..67fb5eb9a5 100644
--- a/testsuite/replay-verify/pyproject.toml
+++ b/testsuite/replay-verify/pyproject.toml
@@ -4,16 +4,17 @@ version = "0.1.0"
 description = ""
 authors = ["Bo Wu <bo@aptoslabs.com>"]
 readme = "README.md"
-packages = [{include = "replay_verify"}]
 
 [tool.poetry.dependencies]
 python = "^3.10"
 kubernetes = "^31.0.0"
 
-
 [tool.poetry.group.dev.dependencies]
 black = "^24.10.0"
 
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.poetry.scripts]
+replay-verify = "main:main"