Minor fix in the automation script