lib/grape/batch.rb in grape-batch-2.0.0 vs lib/grape/batch.rb in grape-batch-2.0.1

@@ -55,10 +55,10 @@
           # init env for Grape resource
           tmp_env = prepare_tmp_env(batch_env, request)
           status, headers, response = @app.call(tmp_env)
 
           # format response
-          @response_klass::format(status, headers, response)
+          @response_klass::format(status, headers, response, tmp_env)
         end
       end
 
       def prepare_tmp_env(tmp_env, request)
         method = request['method']