Fix the bug in test case.