Small fix for Capybara tests