Fix upgrades

Wish github let us rebase on the website