fix test cases.