diff --git a/gem/lib/capybara-select2.rb b/gem/lib/capybara-select2.rb index 9268050..0324552 100644 --- a/gem/lib/capybara-select2.rb +++ b/gem/lib/capybara-select2.rb @@ -17,11 +17,7 @@ def select2(value, options = {}) end # Open select2 field - if select2_container.has_selector?(".select2-choice") - select2_container.find(".select2-choice").click - else - select2_container.find(".select2-choices").click - end + select2_container.first(".select2-choice, .select2-choices").click if options.has_key? :search find(:xpath, "//body").find(".select2-with-searchbox input.select2-input").set(value)