Fix broken "live partitioning change" tests.