Add endpoint handling to upgrades