add missing upgrade step