lib/asir/main.rb in asir-1.2.8 vs lib/asir/main.rb in asir-1.2.9

@@ -375,10 +375,10 @@
     old_arg0 = $0.dup
     after_receive_message = transport.after_receive_message || lambda { | transport, state | nil }
     transport.after_receive_message = lambda do | transport, state |
       message = state.message
       $0 = "#{old_arg0} #{transport.message_count} #{message.identifier}"
-      after_receive_message.call(transport, message)
+      after_receive_message.call(transport, state)
     end
     transport.run_server!
     self
   end