fix test case
fix test cases