A small impr on upgrade task.