diff --git a/git-remote-hg b/git-remote-hg
index adbb82f..ceadec2 100755
--- a/git-remote-hg
+++ b/git-remote-hg
@@ -1595,5 +1595,6 @@ def bye():
     if is_tmp:
         shutil.rmtree(dirname)
 
-atexit.register(bye)
-sys.exit(main(sys.argv))
+if __name__ == '__main__':
+    atexit.register(bye)
+    sys.exit(main(sys.argv))