fix token update bug