Small fix in test case.