lib/rake/funnel/tasks/timing.rb in rake-funnel-0.22.0 vs lib/rake/funnel/tasks/timing.rb in rake-funnel-0.22.1

@@ -52,24 +52,24 @@
 
             p.setup do
               Rake::Application.class_eval do
                 orig_display_error_message = instance_method(:display_error_message)
 
-                define_method(:display_error_message) do |ex|
-                  orig_display_error_message.bind(self).call(ex)
+                define_method(:display_error_message) do |*ex|
+                  orig_display_error_message.bind(self).call(*ex)
 
                   report_invoker.call(failed: true)
                 end
 
                 orig_display_error_message
               end
             end
 
             p.reset do |memo|
               Rake::Application.class_eval do
-                define_method(:display_error_message) do |ex|
-                  memo.bind(self).call(ex)
+                define_method(:display_error_message) do |*ex|
+                  memo.bind(self).call(*ex)
                 end
               end
             end
           end
         end
@@ -80,23 +80,23 @@
 
             p.setup do
               Rake::Task.class_eval do
                 orig_execute = instance_method(:execute)
 
-                define_method(:execute) do |args|
+                define_method(:execute) do |*args|
                   benchmark_invoker.call(self) do
-                    orig_execute.bind(self).call(args)
+                    orig_execute.bind(self).call(*args)
                   end
                 end
 
                 orig_execute
               end
             end
 
             p.reset do |memo|
               Rake::Task.class_eval do
-                define_method(:execute) do |ex|
-                  memo.bind(self).call(ex)
+                define_method(:execute) do |*ex|
+                  memo.bind(self).call(*ex)
                 end
               end
             end
           end
         end