lib/page-object/elements/heading.rb in page-object-0.5.4 vs lib/page-object/elements/heading.rb in page-object-0.5.5

@@ -1,7 +1,15 @@
 module PageObject
   module Elements
     class Heading < Element
 
     end
+
+    ::PageObject::Elements.tag_to_class[:h1] = ::PageObject::Elements::Heading
+    ::PageObject::Elements.tag_to_class[:h2] = ::PageObject::Elements::Heading
+    ::PageObject::Elements.tag_to_class[:h3] = ::PageObject::Elements::Heading
+    ::PageObject::Elements.tag_to_class[:h4] = ::PageObject::Elements::Heading
+    ::PageObject::Elements.tag_to_class[:h5] = ::PageObject::Elements::Heading
+    ::PageObject::Elements.tag_to_class[:h6] = ::PageObject::Elements::Heading
+    
   end
-end
\ No newline at end of file
+end