fix test case
fix test case