diff --git a/libs/vertexai/tests/integration_tests/test_chains.py b/libs/vertexai/tests/integration_tests/test_chains.py
index 0191ebd6..fa10b572 100644
--- a/libs/vertexai/tests/integration_tests/test_chains.py
+++ b/libs/vertexai/tests/integration_tests/test_chains.py
@@ -54,6 +54,7 @@ def test_create_structured_runnable_with_prompt() -> None:
 
 
 @pytest.mark.release
+@pytest.mark.xfail(reason="flaky")
 def test_reflection() -> None:
     class Reflection(BaseModel):
         reflections: str = Field(
diff --git a/libs/vertexai/tests/integration_tests/test_chat_models.py b/libs/vertexai/tests/integration_tests/test_chat_models.py
index 00d6548c..2b5803d8 100644
--- a/libs/vertexai/tests/integration_tests/test_chat_models.py
+++ b/libs/vertexai/tests/integration_tests/test_chat_models.py
@@ -439,8 +439,8 @@ class MyModel(BaseModel):
     }
 
 
-# Can be flaky
 @pytest.mark.release
+@pytest.mark.xfail(reason="flaky")
 def test_chat_vertexai_gemini_function_calling_with_multiple_parts() -> None:
     @tool
     def search(