add upgrade test