fix the test cases