Updated upgrade step