fix the bug of the test case