diff --git a/tests/local_testing/test_custom_callback_input.py b/tests/local_testing/test_custom_callback_input.py
index fcdfcfe6e7..ead387599d 100644
--- a/tests/local_testing/test_custom_callback_input.py
+++ b/tests/local_testing/test_custom_callback_input.py
@@ -1300,9 +1300,11 @@ def test_logging_async_cache_hit_sync_call(turn_off_message_logging):
                 "redacted-by-litellm"
                 == standard_logging_object["messages"][0]["content"]
             )
-            assert {"text": "redacted-by-litellm"} == standard_logging_object[
-                "response"
-            ]
+            # response is a full ModelResponse dict (choices format) since d84e5e381acf
+            assert (
+                standard_logging_object["response"]["choices"][0]["message"]["content"]
+                == "redacted-by-litellm"
+            )
 
 
 def test_logging_standard_payload_failure_call():