Add handling for site upgrades