lib/cloudtasker/batch/middleware/server.rb in cloudtasker-0.13.2 vs lib/cloudtasker/batch/middleware/server.rb in cloudtasker-0.14.rc1

@@ -3,11 +3,11 @@
 module Cloudtasker
   module Batch
     module Middleware
       # Server middleware, invoked when jobs are executed
       class Server
-        def call(worker, **_kwargs)
-          Job.for(worker).execute { yield }
+        def call(worker, **_kwargs, &block)
+          Job.for(worker).execute(&block)
         end
       end
     end
   end
 end