lib/page-object/elements.rb in page-object-0.6.3 vs lib/page-object/elements.rb in page-object-0.6.4

@@ -15,11 +15,11 @@
 
       #
       # method to return the element for a tag_name
       #
       def element_class_for(tag_name, type=nil)
-        return type_to_class[type] if type
+        return type_to_class[type.to_sym] if type
         tag_to_class[tag_name.to_sym] || ::PageObject::Elements::Element
       end
       
     end
   end
@@ -46,7 +46,8 @@
 require 'page-object/elements/unordered_list'
 require 'page-object/elements/ordered_list'
 require 'page-object/elements/option'
 require 'page-object/elements/heading'
 require 'page-object/elements/paragraph'
+require 'page-object/elements/label'
 require 'page-object/elements/file_field'