diff --git a/tests/test_main.py b/tests/test_main.py
index 2dbe47b8f..df1048b1c 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -1,5 +1,6 @@
 import os
 import tempfile
+import unittest
 from unittest import TestCase
 from aider.main import main
 import subprocess