Fix upgrade test case.