Fix broken upgrade test case.