fixes build errors for the upgrade screen