Minor fix for