diff --git a/lib/rails_view_annotator/action_view/partial_renderer.rb b/lib/rails_view_annotator/action_view/partial_renderer.rb
index 7c86663..1a5b72e 100644
--- a/lib/rails_view_annotator/action_view/partial_renderer.rb
+++ b/lib/rails_view_annotator/action_view/partial_renderer.rb
@@ -23,12 +23,12 @@ def self.augment_partial_renderer klass
       if inner.present?
         comment_pattern = "%{partial}"
         template_formats = RailsViewAnnotator.extract_requested_formats_from(args)
-        if template_formats.include?(:text) # Do not render any comments for raw plaintext repsonses
-          return inner
-        elsif template_formats.include?(:js)
+        if template_formats == [:js]
           comment_pattern = "/* begin: %{comment} */\n#{comment_pattern}/* end: %{comment} */"
-        elsif template_formats.empty? || template_formats.include?(:html)
+        elsif template_formats == [:html]
           comment_pattern = "<!-- begin: %{comment} -->\n#{comment_pattern}<!-- end: %{comment} -->"
+        else
+          return inner # Do not render any comments for responses we do not support
         end
 
         (comment_pattern % {:partial => inner, :comment => descriptor}).html_safe