Update setup-stage.sh