diff --git a/tests/test_main.py b/tests/test_main.py
index b21639aff..b4791bf53 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -1,4 +1,5 @@
 import os
+import sys
 import tempfile
 from unittest import TestCase
 from aider.main import main