Fix broken test cases.