Fix broken test cases