diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py
index 4c5c3dc8e0..02683de158 100644
--- a/litellm/tests/test_amazing_vertex_completion.py
+++ b/litellm/tests/test_amazing_vertex_completion.py
@@ -2112,7 +2112,7 @@ def test_get_token_url():
         vertex_credentials=vertex_credentials,
         gemini_api_key="",
         custom_llm_provider="vertex_ai_beta",
-        should_use_vertex_v1beta1_features=should_use_v1beta1_features,
+        should_use_v1beta1_features=should_use_v1beta1_features,
         api_base=None,
         model="",
         stream=False,
@@ -2132,7 +2132,7 @@ def test_get_token_url():
         vertex_credentials=vertex_credentials,
         gemini_api_key="",
         custom_llm_provider="vertex_ai_beta",
-        should_use_vertex_v1beta1_features=should_use_v1beta1_features,
+        should_use_v1beta1_features=should_use_v1beta1_features,
         api_base=None,
         model="",
         stream=False,