diff --git a/rivetkit-typescript/packages/rivetkit/src/workflow/driver.ts b/rivetkit-typescript/packages/rivetkit/src/workflow/driver.ts
index 6bad4e4cd1..e1f4bd4f26 100644
--- a/rivetkit-typescript/packages/rivetkit/src/workflow/driver.ts
+++ b/rivetkit-typescript/packages/rivetkit/src/workflow/driver.ts
@@ -223,9 +223,11 @@ export class ActorWorkflowDriver implements EngineDriver {
 		messageNames: string[],
 		abortSignal: AbortSignal,
 	): Promise<void> {
-		return this.#actor.queueManager.waitForNames(
-			messageNames.length > 0 ? messageNames : undefined,
-			abortSignal,
+		return this.#runCtx.internalKeepAwake(
+			this.#actor.queueManager.waitForNames(
+				messageNames.length > 0 ? messageNames : undefined,
+				abortSignal,
+			),
 		);
 	}
 }