lib/druid/elements/table_row.rb in druid-ts-1.2.2 vs lib/druid/elements/table_row.rb in druid-ts-1.2.3

@@ -1,5 +1,6 @@
+require 'watir/elements/html_elements'
 module Druid
   module Elements
     class TableRow < Element
 
       #
@@ -28,9 +29,12 @@
       private
 
       def find_index_by_title(title)
         table = element.parent
         table = table.parent if table.tag_name == 'tbody'
+        if table.instance_of? Watir::HTMLElement
+          table = table.to_subtype
+        end
         first_row = table[0]
         first_row.cells.find_index { |column| column.text.include? title}
       end
 
     end