Fix bug of test case