Fixing the test cases.