Capybara test improve

@joneslee85