fix the test case