Fixing upgrade step.