diff --git a/tests/image_gen_tests/base_image_generation_test.py b/tests/image_gen_tests/base_image_generation_test.py
index 98c00f305eb..c3a5cfb2251 100644
--- a/tests/image_gen_tests/base_image_generation_test.py
+++ b/tests/image_gen_tests/base_image_generation_test.py
@@ -68,10 +68,15 @@ class BaseImageGenTest(ABC):
             assert logged_standard_logging_payload is not None
             assert logged_standard_logging_payload["response_cost"] is not None
             assert logged_standard_logging_payload["response_cost"] > 0
-
+            import openai
             from openai.types.images_response import ImagesResponse
 
-            response_dict = response.model_dump()
+            # print openai version
+            print("openai version=", openai.__version__)
+
+            response_dict = dict(response)
+            if "usage" in response_dict:
+                response_dict["usage"] = dict(response_dict["usage"])
             print("response usage=", response_dict.get("usage"))
             ImagesResponse.model_validate(response_dict)