Fixes a flaky spec by moving the assertions to the CSS selector so that
Capybara waits for the attributes to be set.