diff --git a/packages/vitest/src/api/setup.ts b/packages/vitest/src/api/setup.ts
index 2742ae27d4d7..abcbe489a26c 100644
--- a/packages/vitest/src/api/setup.ts
+++ b/packages/vitest/src/api/setup.ts
@@ -101,7 +101,7 @@ export function setup(ctx: Vitest) {
       {
         post: msg => ws.send(msg),
         on: fn => ws.on('message', fn),
-        eventNames: ['onCollected', 'onUserConsoleLog', 'onTaskUpdate'],
+        eventNames: ['onUserConsoleLog', 'onFinished', 'onCollected'],
         serialize: stringify,
         deserialize: parse,
       },