Fix the upgrade system.