diff --git a/reload.lua b/reload.lua
index 6e40c2d..e174993 100644
--- a/reload.lua
+++ b/reload.lua
@@ -1,5 +1,16 @@
 local CC_UTILS_DIR = "/cc-utilities"
 
+local function onResume(scriptArgs)
+  if #scriptArgs > 0 then
+    scriptArgs[1] = CC_UTILS_DIR .. scriptArgs[1]
+    shell.run(table.unpack(scriptArgs))
+  end
+end
+
+if pcall(debug.getlocal, 4, 1) then
+  return { onResume = onResume }
+end
+
 local result, retval = pcall(fs.delete, CC_UTILS_DIR)
 if fs.exists(CC_UTILS_DIR) then
   if result then
@@ -11,5 +22,6 @@ if fs.exists(CC_UTILS_DIR) then
 end
 
 shell.run("clone https://gitea.tofvesson.se/GabrielTofvesson/cc-utilities.git")
-shell.run("bg")
-shell.run("cc-utilities/itemcontroller")
\ No newline at end of file
+
+local nextVersion = require(CC_UTILS_DIR .. "reload")
+local _ = ((type(nextVersion) == "table" and type(nextVersion.onResume) == "function") and nextVersion.onResume or onResume)({...})
\ No newline at end of file