X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=lib%2Ferd_handler%2Ferd.rb;h=bf740072d432b883cf7e58d891d56c4431418051;hb=042a27abb3ac3bf9faf36406204fe1d1901123d3;hp=f7e246a0d3581a7c62c747f5575e7956b038d522;hpb=54fb8d2f752d1ad35d11a03bb6e1443257a472c9;p=erd-marker.git

diff --git a/lib/erd_handler/erd.rb b/lib/erd_handler/erd.rb
index f7e246a..bf74007 100644
--- a/lib/erd_handler/erd.rb
+++ b/lib/erd_handler/erd.rb
@@ -10,13 +10,16 @@ module ErdHandler
       doc = Document.new(source)
       raise InvalidErdFile unless doc.elements.to_a.length == 1 and doc.elements[1].name.downcase == 'drawing'
       self.mark = doc.elements['Drawing'].attributes["mark"].to_f
-      self.name = doc.elements['Drawing'].attributes["name"]
+      self.name = Label.new doc.elements['Drawing'].attributes["name"]
       doc.elements.each('Drawing/box') do |box_element|
         self << Box.new(box_element)
       end
       doc.elements.each('Drawing/link') do |link_element|
         self << Link.new(link_element, self.vertices)
       end
+      doc.elements.each('Drawing/selfLink') do |link_element|
+        self << Link.new(link_element, self.vertices)
+      end
       self
     end
   end