lib/tedium/capybara/node_element.rb in tedium-0.0.4 vs lib/tedium/capybara/node_element.rb in tedium-0.0.5

@@ -4,16 +4,15 @@
       def self.included(base)
         base.send(:alias_method, :original_set, :set)
         base.send(:alias_method, :set, :augmented_select)
       end
 
-      def augmented_select(value)
+      def augmented_select(*args)
         if tag_name == "select"
-          find(:option_with_value_or_label, value).select_option
+          find(:option_with_value_or_label, args[0]).select_option
         else
-          original_set(value)
+          original_set(*args)
         end
       end
     end
   end
 end
-