Fix setup script