Add explicit exit code for upgrade