fix broken test cases.