fix test cases issue