Fix load token bug