Minor fix in test suite