Fix upgrade step