lib/core/util/ssh.rb in nucleon-0.2.8 vs lib/core/util/ssh.rb in nucleon-0.2.9

@@ -253,16 +253,16 @@
               unless success
                 raise "Could not execute command: #{command.inspect}"
               end
 
               channel.on_data do |ch, data|
+                data = yield(:output, command, data) if block_given?
                 result.append_output(data)
-                yield(:output, command, data) if block_given?
               end
 
               channel.on_extended_data do |ch, type, data|
+                data = yield(:error, command, data) if block_given?
                 result.append_errors(data)
-                yield(:error, command, data) if block_given?
               end
 
               channel.on_request('exit-status') do |ch, data|
                 result.status = data.read_long
               end