1 files changed, 6 insertions, 1 deletions
diff --git a/runtime/js/99_main.js b/runtime/js/99_main.js
index a66a1660e..e5b9b9778 100644
--- a/runtime/js/99_main.js
+++ b/runtime/js/99_main.js
@@ -294,7 +294,12 @@ async function pollForMessages() {
     );
   }
   while (!isClosing) {
-    const data = await op_worker_recv_message();
+    const recvMessage = op_worker_recv_message();
+    if (globalThis[messagePort.unrefPollForMessages] === true) {
+      core.unrefOpPromise(recvMessage);
+    }
+    const data = await recvMessage;
+    // const data = await op_worker_recv_message();
     if (data === null) break;
     const v = messagePort.deserializeJsMessageData(data);
     const message = v[0];