diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py
index 8bf2ca44e..6aff2be7f 100644
--- a/tests/basic/test_models.py
+++ b/tests/basic/test_models.py
@@ -110,10 +110,10 @@ class TestModels(unittest.TestCase):
         self.assertEqual(model.name, "gpt-3.5-turbo")
 
         model = Model("sonnet")
-        self.assertEqual(model.name, "claude-3-sonnet-20241022")
+        self.assertEqual(model.name, "claude-3-5-sonnet-20241022")
 
         model = Model("haiku")
-        self.assertEqual(model.name, "claude-3-haiku-20241022")
+        self.assertEqual(model.name, "claude-3-5-haiku-20241022")
 
         model = Model("opus")
         self.assertEqual(model.name, "claude-3-opus-20240229")