lib/bullet/rack.rb in bullet-5.0.0 vs lib/bullet/rack.rb in bullet-5.1.0

@@ -65,10 +65,10 @@
     def html_request?(headers, response)
       headers['Content-Type'] && headers['Content-Type'].include?('text/html') && response_body(response).include?("<html")
     end
 
     def response_body(response)
-      if rails?
+      if response.respond_to?(:body)
         Array === response.body ? response.body.first : response.body
       else
         response.first
       end
     end