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