diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py
index f9e79e16a..9c1035d84 100644
--- a/tests/basic/test_models.py
+++ b/tests/basic/test_models.py
@@ -1,9 +1,11 @@
 import unittest
-from unittest.mock import ANY, MagicMock, patch
+from unittest.mock import ANY, MagicMock, patch, call
 
 from aider.models import (
     Model,
     ModelInfoManager,
+    ModelSettings,
+    MODEL_SETTINGS,
     sanity_check_model,
     sanity_check_models,
 )