Fix call to the wrong upgrade step function