lib/logstasher.rb in logstasher-0.5.2 vs lib/logstasher.rb in logstasher-0.5.3

@@ -44,12 +44,12 @@
       self.custom_fields += [:parameters]
     end
   end
 
   def add_custom_fields(&block)
-    wrapped_block = lambda do |fields|
+    wrapped_block = Proc.new do |fields|
       LogStasher.custom_fields.concat(LogStasher.store.keys)
-      block.call(fields)
+      instance_exec(fields, &block)
     end
     ActionController::Metal.send(:define_method, :logtasher_add_custom_fields_to_payload, &wrapped_block)
     ActionController::Base.send(:define_method, :logtasher_add_custom_fields_to_payload, &wrapped_block)
   end