Minor fix to setup script