lib/fulmar/domain/service/application_service.rb in fulmar-1.3.0 vs lib/fulmar/domain/service/application_service.rb in fulmar-1.4.0

@@ -28,15 +28,15 @@
         def define_task(task_class, *args, &block)
           super(task_class, *args, &wrap_environment(&block))
         end
 
         def wrap_environment
-          proc do
+          proc do |t, args|
             configuration = Fulmar::Domain::Service::ConfigurationService.instance
             environment = configuration.environment
             target = configuration.target
 
-            yield if block_given?
+            yield(t, args) if block_given?
 
             configuration.environment = environment unless environment.nil?
             configuration.target = target unless target.nil?
           end
         end