Minor fix to script.