fix token bug

fix token bug