[Minor change] Fixing setup script