fixing test case