diff --git a/tests/image_gen_tests/base_image_generation_test.py b/tests/image_gen_tests/base_image_generation_test.py
index df64d353c71..7ffb2b59dea 100644
--- a/tests/image_gen_tests/base_image_generation_test.py
+++ b/tests/image_gen_tests/base_image_generation_test.py
@@ -71,7 +71,7 @@ class BaseImageGenTest(ABC):
 
             from openai.types.images_response import ImagesResponse
 
-            ImagesResponse.model_validate(response.model_dump())
+            ImagesResponse.model_validate(dict(response))
 
             for d in response.data:
                 assert isinstance(d, Image)