1 files changed, 7 insertions, 2 deletions
diff --git a/cli/tests/workers_startup_bench.ts b/cli/tests/workers_startup_bench.ts
index 5d2c24b89..60c15a4b1 100644
--- a/cli/tests/workers_startup_bench.ts
+++ b/cli/tests/workers_startup_bench.ts
@@ -4,7 +4,7 @@ const workerCount = 50;
 async function bench(): Promise<void> {
   const workers: Worker[] = [];
   for (let i = 1; i <= workerCount; ++i) {
-    const worker = new Worker("./subdir/bench_worker.ts");
+    const worker = new Worker("./subdir/bench_worker.ts", { type: "module" });
     const promise = new Promise((resolve): void => {
       worker.onmessage = (e): void => {
         if (e.data.cmdId === 0) resolve();
@@ -16,8 +16,13 @@ async function bench(): Promise<void> {
   }
   console.log("Done creating workers closing workers!");
   for (const worker of workers) {
+    const promise = new Promise((resolve): void => {
+      worker.onmessage = (e): void => {
+        if (e.data.cmdId === 3) resolve();
+      };
+    });
     worker.postMessage({ action: 3 });
-    await worker.closed; // Required to avoid a cmdId not in table error.
+    await promise;
   }
   console.log("Finished!");
 }