Fix bug in test case.