crates/dtmm/assets/mod_main.lua

@@ -59,8 +59,13 @@ local function patch_mod_loading_state()
     print("[mod_main] Adding mod loading state")
     local GameStateMachine = require("scripts/foundation/utilities/game_state_machine")
 
+    local patched = false
+
     local GameStateMachine_init = GameStateMachine.init
     GameStateMachine.init = function(self, parent, start_state, params, ...)
+        if not patched then
+            patched = true
+
             -- Hardcoded position after `StateRequireScripts`.
             -- We do want to wait until then, so that most of the game's core
             -- systems are at least loaded and can be hooked, even if they aren't
@@ -72,12 +77,11 @@ print("[mod_main] Adding mod loading state")
                     package_manager = params.package_manager,
                 },
             })
-
-        -- Clean up after us
-        GameStateMachine.init = GameStateMachine_init
-
-        return GameStateMachine_init(self, parent, start_state, params, ...)
         end
+
+        GameStateMachine_init(self, parent, start_state, params, ...)
+    end
+    print("[mod_main] Mod patching complete")
 end
 
 function init()