diff --git a/aider/main.py b/aider/main.py
index bb837ef43..e6b417250 100644
--- a/aider/main.py
+++ b/aider/main.py
@@ -17,7 +17,7 @@ from aider.history import ChatSummary
 from aider.io import InputOutput
 from aider.llm import litellm  # noqa: F401; properly init litellm on launch
 from aider.repo import GitRepo
-from aider.versioncheck import check_version, install_from_main_branch
+from aider.versioncheck import check_version, install_from_main_branch, install_upgrade
 
 from .dump import dump  # noqa: F401