trying to fix test case