class Bullet::Rack

def response_body(response)
  if response.respond_to?(:body)
    Array === response.body ? response.body.first : response.body
  elsif response.respond_to?(:first)
    response.first
  end
end