Update selectors in test helpers