fixing test cases
test case fix