fix test code
match test case