diff --git a/src/test/java/com/box/sdk/BoxAITest.java b/src/test/java/com/box/sdk/BoxAITest.java
index 7607f9088..0217c83ea 100644
--- a/src/test/java/com/box/sdk/BoxAITest.java
+++ b/src/test/java/com/box/sdk/BoxAITest.java
@@ -360,7 +360,7 @@ public void testExtractMetadataStructuredSuccess() {
             agent
         );
 
-        assertThat(response, equalTo(Json.parse(result).asObject()));
+        assertThat(response.getSourceJson(), equalTo(Json.parse(result).asObject()));
         assertThat(response.getSourceJson().get("firstName").asString(), equalTo("John"));
     }
 }