diff --git a/litellm/tests/test_exceptions.py b/litellm/tests/test_exceptions.py
index 9c90014c0d6..311bbfa5716 100644
--- a/litellm/tests/test_exceptions.py
+++ b/litellm/tests/test_exceptions.py
@@ -70,7 +70,7 @@ models = ["command-nightly"]
 @pytest.mark.parametrize("model", models)
 def test_context_window_with_fallbacks(model):
     ctx_window_fallback_dict = {
-        "command-nightly": "claude-2",
+        "command-nightly": "claude-2.1",
         "gpt-3.5-turbo-instruct": "gpt-3.5-turbo-16k",
         "azure/chatgpt-v-2": "gpt-3.5-turbo-16k",
     }