lib/logstasher/rails_ext/action_controller/metal/instrumentation.rb in logstasher-1.0.0 vs lib/logstasher/rails_ext/action_controller/metal/instrumentation.rb in logstasher-1.0.1

@@ -26,10 +26,10 @@
         if self.respond_to?(:logtasher_add_custom_fields_to_payload)
           before_keys = raw_payload.keys.clone
           logtasher_add_custom_fields_to_payload(raw_payload)
           after_keys = raw_payload.keys
           # Store all extra keys added to payload hash in payload itself. This is a thread safe way
-          LogStasher.custom_fields += after_keys - before_keys
+          LogStasher::CustomFields.add(*(after_keys - before_keys))
         end
 
         result = super
 
         payload[:status] = response.status