diff --git a/packages/app/e2e/session/session-review.spec.ts b/packages/app/e2e/session/session-review.spec.ts
index b7695cc0b5..6e7df88eed 100644
--- a/packages/app/e2e/session/session-review.spec.ts
+++ b/packages/app/e2e/session/session-review.spec.ts
@@ -1,6 +1,6 @@
 import { waitSessionIdle, withSession } from "../actions"
 import { test, expect } from "../fixtures"
-import { inputMatch } from "../prompt/mock"
+import { promptMatch } from "../prompt/mock"
 
 const count = 14
 
@@ -47,7 +47,7 @@ async function patchWithMock(
   patchText: string,
 ) {
   const callsBefore = await llm.calls()
-  await llm.toolMatch(inputMatch({ patchText }), "apply_patch", { patchText })
+  await llm.toolMatch(promptMatch("Apply the provided patch exactly once."), "apply_patch", { patchText })
   await sdk.session.promptAsync({
     sessionID,
     agent: "build",