Minor fix in unit test.