Fixed the tokeninput bug