Fix token update