lib/grape/middleware/base.rb in grape-1.6.0 vs lib/grape/middleware/base.rb in grape-1.6.1

@@ -58,10 +58,10 @@
       def after; end
 
       def response
         return @app_response if @app_response.is_a?(Rack::Response)
 
-        Rack::Response.new(@app_response[2], @app_response[0], @app_response[1])
+        @app_response = Rack::Response.new(@app_response[2], @app_response[0], @app_response[1])
       end
 
       def content_type_for(format)
         HashWithIndifferentAccess.new(content_types)[format]
       end