fixing the testcase