Fix for Capybara 2