fix upgrade test