lib/plantuml_builder/formats.rb in plantuml_builder-0.1.1 vs lib/plantuml_builder/formats.rb in plantuml_builder-0.2.0

@@ -12,10 +12,12 @@
       def format
         'svg'
       end
 
       def make_diagram_versionable
-        self.data = data.split('>').join(">\n").gsub(/<!--.*-->/m, '')
+        self.data = data.split('>')
+                        .join(">\n")
+                        .gsub(/<!--[\s\S]*?-->/, '') + '>'
       end
     end
 
     class PNG < PlantumlBuilder::Format
       protected
\ No newline at end of file