Quick change to setup script