diff --git a/src/agents/pi-hooks/context-pruning.test.ts b/src/agents/pi-hooks/context-pruning.test.ts
index cf65cdc0d35..a2da89bb330 100644
--- a/src/agents/pi-hooks/context-pruning.test.ts
+++ b/src/agents/pi-hooks/context-pruning.test.ts
@@ -405,7 +405,7 @@ describe("context-pruning", () => {
     });
 
     const tool = findToolResult(next, "t1");
-    expect(tool.content.some((b) => b.type === "image")).toBe(false);
+    expect(tool.content.filter((block) => block.type === "image")).toEqual([]);
     expect(toolText(tool)).toContain("[image removed during context pruning]");
     expect(toolText(tool)).toContain("visible tool text");
   });