Fixed up a test, reused some selectors threw error