fixing broken test case