lib/xray-rails.rb in xray-rails-0.1.10 vs lib/xray-rails.rb in xray-rails-0.1.11

@@ -79,9 +79,12 @@
   #     ...
   #   </div>
   #   <!-- XRAY END 123 -->
   def self.augment_template(source, path)
     id = next_id
+    if source.include?('<!DOCTYPE')
+      return source
+    end
     # skim doesn't allow html comments, so use skim's comment syntax if it's skim
     if path =~ /\.(skim)(\.|$)/
       augmented = "/!XRAY START #{id} #{path}\n#{source}\n/!XRAY END #{id}"
     else
       augmented = "<!--XRAY START #{id} #{path}-->\n#{source}\n<!--XRAY END #{id}-->"