fixing for the testcase