lib/helium/console/registry.rb in helium-console-0.1.9 vs lib/helium/console/registry.rb in helium-console-0.1.10

@@ -66,9 +66,10 @@
       end
 
       def handler_for(object, **options)
         element_class = object.class.ancestors.find do |ancestor|
           break handlers[ancestor] if handlers.key?(ancestor)
+          break handlers[ancestor.name] if handlers.key?(ancestor.name)
         end
         return unless element_class
 
         element_class.new(object, **options)
       end