Prepare for the upgrade