small modification to setup script