diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py
index 53bb9fd8030..bad2428fbe2 100644
--- a/litellm/tests/test_amazing_vertex_completion.py
+++ b/litellm/tests/test_amazing_vertex_completion.py
@@ -1488,6 +1488,9 @@ async def test_gemini_pro_json_schema_args_sent_httpx_openai_schema(
 ):
     from typing import List
 
+    if enforce_validation:
+        litellm.enable_json_schema_validation = True
+
     from pydantic import BaseModel
 
     load_vertex_ai_credentials()