diff --git a/python/pip_install/extract_wheels/lib/bazel.py b/python/pip_install/extract_wheels/lib/bazel.py
index 433a9ea54a..1ea4a7b77a 100644
--- a/python/pip_install/extract_wheels/lib/bazel.py
+++ b/python/pip_install/extract_wheels/lib/bazel.py
@@ -35,8 +35,7 @@ def generate_entry_point_contents(entry_point: str, shebang: str = "#!/usr/bin/e
         from {module} import {method}
         if __name__ == "__main__":
             rc = {method}()
-            if rc.is_integer():
-                sys.exit(rc)
+            sys.exit({method}())
         """.format(
         shebang=shebang,
         module=module,
diff --git a/python/pip_install/extract_wheels/lib/bazel_test.py b/python/pip_install/extract_wheels/lib/bazel_test.py
index 97bc06e779..5c22047479 100644
--- a/python/pip_install/extract_wheels/lib/bazel_test.py
+++ b/python/pip_install/extract_wheels/lib/bazel_test.py
@@ -10,9 +10,7 @@ def test_generate_entry_point_contents(self):
 import sys
 from sphinx.cmd.build import main
 if __name__ == "__main__":
-    rc = main()
-    if rc.is_integer():
-        sys.exit(rc)
+    sys.exit(main())
 """
         self.assertEqual(got, want)
 
@@ -22,8 +20,6 @@ def test_generate_entry_point_contents_with_shebang(self):
         want = """#!/usr/bin/python
 import sys
 from sphinx.cmd.build import main
-rc = main()
-if rc.is_integer():
-    sys.exit(rc)
+sys.exit(main())
 """
         self.assertEqual(got, want)