Update the upgrade test for dmg