lib/potassium/templates/application/helpers/callback-helpers.rb in potassium-1.3.4 vs lib/potassium/templates/application/helpers/callback-helpers.rb in potassium-1.3.5

@@ -9,11 +9,11 @@
     end
   end
 
   def before(action_name, wrap_in_action: false, &action)
     ensure_callbacks_variables_for_action(action_name)
-    add_callback(action_name, :before, action)
+    add_callback(action_name, :before, wrap_action(action, wrap_in_action))
   end
 
   def run_action(action_name, &action)
     ensure_callbacks_variables_for_action(action_name)
     callbacks = get(:callbacks)[action_name]
@@ -30,10 +30,10 @@
     get(:callbacks)[action_name][type] << action
   end
 
   def wrap_action(action, wrap_in_action)
     return action unless wrap_in_action
-    ->{ run_action(wrap_in_action, &action) }
+    -> { run_action(wrap_in_action, &action) }
   end
 
   def ensure_callbacks_variables_for_action(action_name)
     ensure_variable(:callbacks, {})
     ensure_variable(:executed_actions, [])