diff --git a/src/NetscriptFunctions.js b/src/NetscriptFunctions.js
index aa4c877dc..cbc0c6162 100644
--- a/src/NetscriptFunctions.js
+++ b/src/NetscriptFunctions.js
@@ -1139,7 +1139,7 @@ function NetscriptFunctions(workerScript) {
                     var oldScript = destServer.scripts[i];
                     oldScript.code = sourceScript.code;
                     oldScript.ramUsage = sourceScript.ramUsage;
-                    oldScript.module = "";
+                    oldScript.module.markUpdated();
                     return true;
                 }
             }
diff --git a/src/NetscriptWorker.js b/src/NetscriptWorker.js
index 319693358..095ba401d 100644
--- a/src/NetscriptWorker.js
+++ b/src/NetscriptWorker.js
@@ -534,7 +534,7 @@ export function loadAllRunningScripts() {
 
             // Reset modules on all scripts
             for (let i = 0; i < server.scripts.length; ++i) {
-                server.scripts[i].module = "";
+                server.scripts[i].markUpdated();
             }
 
             if (skipScriptLoad) {