diff --git a/babi/__main__.py b/babi/__main__.py
index ff09674..c1533a7 100644
--- a/babi/__main__.py
+++ b/babi/__main__.py
@@ -3,4 +3,4 @@ from __future__ import annotations
 from babi.main import main
 
 if __name__ == '__main__':
-    exit(main())
+    raise SystemExit(main())
diff --git a/babi/main.py b/babi/main.py
index 959c9a6..bdae492 100644
--- a/babi/main.py
+++ b/babi/main.py
@@ -152,4 +152,4 @@ def main(argv: Sequence[str] | None = None) -> int:
 
 
 if __name__ == '__main__':
-    exit(main())
+    raise SystemExit(main())
diff --git a/babi/textmate_demo.py b/babi/textmate_demo.py
index 2945ff6..8eb696a 100644
--- a/babi/textmate_demo.py
+++ b/babi/textmate_demo.py
@@ -67,4 +67,4 @@ def main(argv: Sequence[str] | None = None) -> int:
 
 
 if __name__ == '__main__':
-    exit(main())
+    raise SystemExit(main())