diff --git a/aider/getinput.py b/aider/getinput.py
index 75431d8d4..70257f299 100644
--- a/aider/getinput.py
+++ b/aider/getinput.py
@@ -3,7 +3,7 @@ from pygments.lexers import guess_lexer_for_filename
 from pygments.token import Token
 from prompt_toolkit.styles import Style
 from pygments.util import ClassNotFound
-from prompt_toolkit.shortcuts import PromptSession
+from prompt_toolkit.shortcuts import PromptSession, prompt
 from prompt_toolkit.completion import Completer, Completion
 from prompt_toolkit.history import FileHistory
 from prompt_toolkit.shortcuts import CompleteStyle