Fix broken upgrade.