Use the updated query selector API