diff --git a/aider/main.py b/aider/main.py
index 1bc2e6f5d..ac5eb19e7 100644
--- a/aider/main.py
+++ b/aider/main.py
@@ -18,6 +18,7 @@ from aider import __version__, models, urls, utils
 from aider.analytics import Analytics
 from aider.args import get_parser
 from aider.coders import Coder
+from aider.coders.base_coder import UnknownEditFormat
 from aider.commands import Commands, SwitchCoder
 from aider.format_settings import format_settings, scrub_sensitive_info
 from aider.history import ChatSummary