Fix capybara tests