Update test for new css selector