Fix formatting in upgrade