Fix token bug