diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py
index d49649739..8d75d9e68 100644
--- a/tests/basic/test_models.py
+++ b/tests/basic/test_models.py
@@ -1,7 +1,7 @@
 import unittest
 from unittest.mock import MagicMock, patch
 
-from aider.models import Model, get_model_info, sanity_check_models
+from aider.models import Model, get_model_info, sanity_check_model, sanity_check_models
 
 
 class TestModels(unittest.TestCase):