diff --git a/tests/test_agent.py b/tests/test_agent.py
index abfab84..c2de3fd 100644
--- a/tests/test_agent.py
+++ b/tests/test_agent.py
@@ -156,8 +156,8 @@ async def test_end_to_end_deterministic_crashing_image_scenario(ctx, workspace):
         affected_services=["checkout"],
         diagnostic_targets=DiagnosticTargets(),
     )
-    assert diagnosis.status == "diagnosed"
-    assert "crashing" in diagnosis.likely_root_cause.lower()
+    assert diagnosis["status"] == "diagnosed"
+    assert "crashing" in diagnosis["likely_root_cause"].lower()
     assert workspace.exists("outputs/incidents/checkout-001/diagnosis.json")
 
     plan = await agent.plan_remediation(ctx, "checkout-001", "rollback the deployment to the previous image")