token bug fix