diff --git a/src/agents/pi-embedded-runner/run/attempt.model-diagnostic-events.ts b/src/agents/pi-embedded-runner/run/attempt.model-diagnostic-events.ts
index 339433c6561..c2751106d87 100644
--- a/src/agents/pi-embedded-runner/run/attempt.model-diagnostic-events.ts
+++ b/src/agents/pi-embedded-runner/run/attempt.model-diagnostic-events.ts
@@ -76,7 +76,7 @@ function baseModelCallEvent(
 }
 
 async function safeReturnIterator(iterator: AsyncIterator<unknown>): Promise<void> {
-  let returnResult: PromiseLike<unknown> | unknown;
+  let returnResult: unknown;
   try {
     returnResult = iterator.return?.();
   } catch {