fixing testcases