diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py
index 6e20e956068..61412cc9189 100644
--- a/tests/test_litellm/test_utils.py
+++ b/tests/test_litellm/test_utils.py
@@ -42,6 +42,8 @@ def test_get_optional_params_image_gen():
     )
     assert optional_params is not None
     assert "response_format" not in optional_params
+    if 'extra_body' in optional_params:
+        assert "response_format" not in optional_params['extra_body']
     assert optional_params["n"] == 3