diff --git a/src/Netscript/killWorkerScript.ts b/src/Netscript/killWorkerScript.ts index 27814340a..9b1850054 100644 --- a/src/Netscript/killWorkerScript.ts +++ b/src/Netscript/killWorkerScript.ts @@ -66,11 +66,12 @@ function stopAndCleanUpWorkerScript(ws: WorkerScript): void { if (typeof ws.atExit === "function") { try { ws.env.stopFlag = false; - ws.atExit(); + const atExit = ws.atExit; + ws.atExit = undefined; + atExit(); } catch (e: unknown) { handleUnknownError(e, ws, "Error running atExit function.\n\n"); } - ws.atExit = undefined; } ws.env.stopFlag = true; removeWorkerScript(ws);