diff --git a/challenges/2023/02-cubeConundrum/main.py b/challenges/2023/02-cubeConundrum/main.py
index edd1234..863d713 100644
--- a/challenges/2023/02-cubeConundrum/main.py
+++ b/challenges/2023/02-cubeConundrum/main.py
@@ -1,4 +1,3 @@
-import os
 import sys
 from dataclasses import dataclass
 
@@ -82,7 +81,7 @@ def _debug(*args, **kwargs):
 if __name__ == "__main__":
     if len(sys.argv) < 2 or sys.argv[1] not in ["1", "2"]:
         print("Missing day argument", file=sys.stderr)
-        os.exit(1)
+        sys.exit(1)
     inp = sys.stdin.read().strip()
     if sys.argv[1] == "1":
         print(one(inp))
diff --git a/challenges/2023/03-gearRatios/main.py b/challenges/2023/03-gearRatios/main.py
index a84ac53..4719b9c 100644
--- a/challenges/2023/03-gearRatios/main.py
+++ b/challenges/2023/03-gearRatios/main.py
@@ -1,4 +1,3 @@
-import os
 import sys
 
 
@@ -127,7 +126,7 @@ def two(inp: str):
 if __name__ == "__main__":
     if len(sys.argv) < 2 or sys.argv[1] not in ["1", "2"]:
         print("Missing day argument", file=sys.stderr)
-        os.exit(1)
+        sys.exit(1)
     inp = sys.stdin.read().strip()
     if sys.argv[1] == "1":
         print(one(inp))
diff --git a/templates/main.py b/templates/main.py
index 668a1d0..5d013d3 100644
--- a/templates/main.py
+++ b/templates/main.py
@@ -1,4 +1,3 @@
-import os
 import sys
 
 
@@ -18,7 +17,7 @@ def two(inp: str):
 if __name__ == "__main__":
     if len(sys.argv) < 2 or sys.argv[1] not in ["1", "2"]:
         print("Missing day argument", file=sys.stderr)
-        os.exit(1)
+        sys.exit(1)
     inp = sys.stdin.read().strip()
     if sys.argv[1] == "1":
         print(one(inp))