Fix broken Capybara tests