Fixing the test case.