diff --git a/src/channels/run-state-machine.test.ts b/src/channels/run-state-machine.test.ts
index a46a5081ab8..2bcf3d9f199 100644
--- a/src/channels/run-state-machine.test.ts
+++ b/src/channels/run-state-machine.test.ts
@@ -16,13 +16,11 @@ describe("createRunStateMachine", () => {
     });
     machine.onRunStart();
     machine.onRunEnd();
-    expect(setStatus).toHaveBeenNthCalledWith(
-      2,
-      expect.objectContaining({ activeRuns: 1, busy: true, lastRunActivityAt: 123 }),
-    );
-    expect(setStatus).toHaveBeenLastCalledWith(
-      expect.objectContaining({ activeRuns: 0, busy: false, lastRunActivityAt: 123 }),
-    );
+    expect(setStatus.mock.calls).toEqual([
+      [{ activeRuns: 0, busy: false }],
+      [{ activeRuns: 1, busy: true, lastRunActivityAt: 123 }],
+      [{ activeRuns: 0, busy: false, lastRunActivityAt: 123 }],
+    ]);
   });
 
   it("stops publishing after lifecycle abort", () => {