Minor fix in script