fixing test case working