Minor fix for test