Add missing XML in upgrade step