diff --git a/tests/benchmarks/bm_main.py b/tests/benchmarks/bm_main.py
index 88e0585e..ce10012e 100755
--- a/tests/benchmarks/bm_main.py
+++ b/tests/benchmarks/bm_main.py
@@ -12,7 +12,7 @@ import sys
 from os.path import dirname, isfile, join
 
 
-if __name__ == "__main__":
+def main() -> None:
     # pyre-ignore[16]
     if len(sys.argv) > 1:
         # Parse from flags.
@@ -36,3 +36,7 @@ if __name__ == "__main__":
     os.environ["PYTHONPATH"] = ":".join(sys.path)
     for file_name in file_names:
         subprocess.check_call([sys.executable, file_name])
+
+
+if __name__ == "__main__":
+    main()  # pragma: no cover