fix token bug