Fixing testcase