FIx the test cases.