Minor fix in setup script