Fix token bugs