Fix issue with test case