Minor fix in test file