Fixes Capybara tests