Fix missing add token on click when only one element