Fix my typo on computeNextToken