Upgrade the stages object