lib/page-object/elements/table.rb in page-object-0.2.5 vs lib/page-object/elements/table.rb in page-object-0.3.0

@@ -25,17 +25,17 @@
         ".//child::tr"
       end
 
       def include_platform_for platform
         super
-        if platform[:platform] == :watir
-          require 'page-object/platforms/watir/table'
-          self.class.send :include, PageObject::Platforms::Watir::Table
-        elsif platform[:platform] == :selenium
-          require 'page-object/platforms/selenium/table'
-          self.class.send :include, PageObject::Platforms::Selenium::Table
+        if platform[:platform] == :watir_webdriver
+          require 'page-object/platforms/watir_webdriver/table'
+          self.class.send :include, PageObject::Platforms::WatirWebDriver::Table
+        elsif platform[:platform] == :selenium_webdriver
+          require 'page-object/platforms/selenium_webdriver/table'
+          self.class.send :include, PageObject::Platforms::SeleniumWebDriver::Table
         else
-          raise ArgumentError, "expect platform to be :watir or :selenium"
+          raise ArgumentError, "expect platform to be :watir_webdriver or :selenium_webdriver"
         end
       end
     end
   end
 end
\ No newline at end of file