Change upgrade setup for leveling