I like what you're testing here, but feel like there's a better way to go about it. Let's look up some ways of using capybara to test for the right class.