diff --git a/litellm/tests/test_tpm_rpm_routing_v2.py b/litellm/tests/test_tpm_rpm_routing_v2.py
index e362eedb0ea..1f3de09104f 100644
--- a/litellm/tests/test_tpm_rpm_routing_v2.py
+++ b/litellm/tests/test_tpm_rpm_routing_v2.py
@@ -394,7 +394,7 @@ async def test_router_completion_streaming():
                 "model": "azure/gpt-turbo",
                 "api_key": "os.environ/AZURE_FRANCE_API_KEY",
                 "api_base": "https://openai-france-1234.openai.azure.com",
-                "rpm": 1440,
+                "tpm": 1440,
                 "mock_response": "Hello world",
             },
             "model_info": {"id": 1},
@@ -405,7 +405,7 @@ async def test_router_completion_streaming():
                 "model": "azure/gpt-35-turbo",
                 "api_key": "os.environ/AZURE_EUROPE_API_KEY",
                 "api_base": "https://my-endpoint-europe-berri-992.openai.azure.com",
-                "rpm": 6,
+                "tpm": 6,
                 "mock_response": "Hello world",
             },
             "model_info": {"id": 2},