Fix test to make sure it picks up the right token