Fix issue in test case.