Minor fix for spec