Fix setup script
Fix setup script