diff --git a/tests/test_main.py b/tests/test_main.py
index a4116679b..2645c2ea8 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -21,9 +21,9 @@ class TestMain(TestCase):
         self.tempdir = tempfile.mkdtemp()
         os.chdir(self.tempdir)
 
-        with patch("aider.coder.Coder.check_for_model_availability") as mock_check:
-            mock_check.return_value = True
-            self.mock_check = mock_check
+    with patch("aider.coder.Coder.check_model_availability") as mock_check:
+        mock_check.return_value = True
+        self.mock_check = mock_check
 
     def tearDown(self):
         os.chdir(self.original_cwd)