Fix incorrect selectors in tests