diff --git a/libs/partners/openai/tests/unit_tests/chat_models/test_base.py b/libs/partners/openai/tests/unit_tests/chat_models/test_base.py
index e1d83a4f047..0563d56c286 100644
--- a/libs/partners/openai/tests/unit_tests/chat_models/test_base.py
+++ b/libs/partners/openai/tests/unit_tests/chat_models/test_base.py
@@ -2532,11 +2532,7 @@ def test_convert_to_v1_from_chat_completions(
                             },
                         ],
                     },
-                    {
-                        "type": "image",
-                        "base64": "...",
-                        "id": "img_123",
-                    },
+                    {"type": "image", "base64": "...", "id": "img_123"},
                     {
                         "type": "non_standard",
                         "value": {"type": "something_else", "foo": "bar"},
@@ -2736,11 +2732,7 @@ def test_convert_from_v1_to_responses(
                             },
                         ],
                     },
-                    {
-                        "type": "image",
-                        "base64": "...",
-                        "id": "img_123",
-                    },
+                    {"type": "image", "base64": "...", "id": "img_123"},
                     {
                         "type": "non_standard",
                         "value": {"type": "something_else", "foo": "bar"},