diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py
index 3c2e0ddda..f04e6671e 100644
--- a/tests/basic/test_coder.py
+++ b/tests/basic/test_coder.py
@@ -7,6 +7,7 @@ from unittest.mock import MagicMock, patch
 import git
 
 from aider.coders import Coder
+from aider.coders.base_coder import UnknownEditFormat
 from aider.dump import dump  # noqa: F401
 from aider.io import InputOutput
 from aider.models import Model