diff --git a/packages/core/src/PageAgentCore.test.ts b/packages/core/src/PageAgentCore.test.ts
index 0518752..5dc8010 100644
--- a/packages/core/src/PageAgentCore.test.ts
+++ b/packages/core/src/PageAgentCore.test.ts
@@ -297,18 +297,8 @@ describe.concurrent('PageAgentCore lifecycle', () => {
 		})
 
 		it('re-throws and sets error status when onAfterStep throws', async () => {
-			// `done` breaks before onAfterStep, so use a non-terminal action.
-			const fetchMock = createFetchMock().mockResolvedValueOnce(
-				agentResponse({ action: { noop: {} } })
-			)
+			const fetchMock = createFetchMock().mockResolvedValueOnce(doneResponse('all done'))
 			const agent = createAgent(fetchMock, {
-				customTools: {
-					noop: tool({
-						description: 'No-op.',
-						inputSchema: z.object({}),
-						execute: async () => 'ok',
-					}),
-				},
 				onAfterStep: async () => {
 					throw new Error('after step failed')
 				},