lib/page-object/elements/table.rb in page-object-0.3.2 vs lib/page-object/elements/table.rb in page-object-0.4.0

@@ -1,5 +1,7 @@
+require 'mixology'
+
 module PageObject
   module Elements
     class Table < Element
       include Enumerable
 
@@ -27,13 +29,13 @@
 
       def include_platform_for platform
         super
         if platform[:platform] == :watir_webdriver
           require 'page-object/platforms/watir_webdriver/table'
-          self.class.send :include, PageObject::Platforms::WatirWebDriver::Table
+          self.mixin PageObject::Platforms::WatirWebDriver::Table
         elsif platform[:platform] == :selenium_webdriver
           require 'page-object/platforms/selenium_webdriver/table'
-          self.class.send :include, PageObject::Platforms::SeleniumWebDriver::Table
+          self.mixin PageObject::Platforms::SeleniumWebDriver::Table
         else
           raise ArgumentError, "expect platform to be :watir_webdriver or :selenium_webdriver"
         end
       end
     end
\ No newline at end of file