diff --git a/tests/test_litellm/llms/vertex_ai/test_vertex.py b/tests/test_litellm/llms/vertex_ai/test_vertex.py
index 803584b5615..bd12100a88f 100644
--- a/tests/test_litellm/llms/vertex_ai/test_vertex.py
+++ b/tests/test_litellm/llms/vertex_ai/test_vertex.py
@@ -393,6 +393,7 @@ def test_multiple_function_call():
                     ],
                 },
                 {
+                    "role": "user",
                     "parts": [
                         {
                             "function_response": {
@@ -498,6 +499,7 @@ def test_multiple_function_call_changed_text_pos():
                 ],
             },
             {
+                "role": "user",
                 "parts": [
                     {
                         "function_response": {