fix test case
fixing tests