diff --git a/common/media/lua/client/TOC/Controllers/LimitActionsController.lua b/common/media/lua/client/TOC/Controllers/LimitActionsController.lua
index d118140..5583217 100644
--- a/common/media/lua/client/TOC/Controllers/LimitActionsController.lua
+++ b/common/media/lua/client/TOC/Controllers/LimitActionsController.lua
@@ -38,17 +38,12 @@ end
 
 -- We need to override when the player changes key binds manually to be sure that TOC changes are re-applied
 local og_MainOptions_apply = MainOptions.apply
-function MainOptions:apply()
-    og_MainOptions_apply(self)
+function MainOptions:apply(closeAfter)
+    og_MainOptions_apply(self, closeAfter)
     CachedDataHandler.OverrideBothHandsFeasibility()
 end
 
 
-
-
-
-
-
 --------------------------------------------
 --* TIMED ACTIONS 
 --* We want to be able to modify how long actions are gonna take,