diff --git a/extensions/matrix/src/matrix/monitor/handler.test.ts b/extensions/matrix/src/matrix/monitor/handler.test.ts
index 452845098258..d2ad4a1dad76 100644
--- a/extensions/matrix/src/matrix/monitor/handler.test.ts
+++ b/extensions/matrix/src/matrix/monitor/handler.test.ts
@@ -4083,7 +4083,9 @@ describe("matrix monitor handler draft streaming", () => {
       }
       await finish();
 
-      const redactedEventIds = redactEventMock.mock.calls.map(([, eventId]) => eventId);
+      const redactedEventIds = mockCalls(redactEventMock, "redactEvent").map(
+        ([, eventId]) => eventId,
+      );
       expect(redactedEventIds.filter((eventId) => eventId === "$draft1")).toHaveLength(
         priorDisposition === "consumed" ? 1 : 0,
       );