Fix generate token bug