Token input fix