diff --git a/tests/core/test_core_tpu.py b/tests/core/test_core_tpu.py
index 8b4568a1c..82806f63b 100644
--- a/tests/core/test_core_tpu.py
+++ b/tests/core/test_core_tpu.py
@@ -202,7 +202,7 @@ def test_utility_method_dispatch(self):
         self.assertIsInstance(output, EngineCoreOutputs)
         self.assertIsInstance(output.utility_output, UtilityOutput)
         self.assertEqual(output.utility_output.call_id, "call-id-1")
-        self.assertEqual(output.utility_output.result, {1, 2, 3})
+        self.assertEqual(output.utility_output.result.result, {1, 2, 3})
 
     def test_prefill_logic(self):
         """Tests the core logic of the _prefill method for one iteration."""