diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py
index 77ed041b3..6071e6c40 100644
--- a/tests/basic/test_models.py
+++ b/tests/basic/test_models.py
@@ -1,5 +1,5 @@
 import unittest
-from unittest.mock import MagicMock, patch
+from unittest.mock import ANY, MagicMock, patch
 
 from aider.models import Model, get_model_info, sanity_check_model, sanity_check_models