diff --git a/setup/so-setup b/setup/so-setup
index ebce5861e..e1177c21c 100755
--- a/setup/so-setup
+++ b/setup/so-setup
@@ -58,7 +58,6 @@ function progress() {
 		fi
 
 		export SO_ERROR=1
-		export SKIP_REBOOT=1
 		title="Error found, please check $setup_log"
 	fi
 
@@ -610,7 +609,7 @@ fi
 	checkin_at_boot >> $setup_log 2>&1
 
 	set_progress_str 95 'Verifying setup'
-	salt-call -l info  state.highstate >> $setup_log 2>&1
+	salt-call -l info state.highstate >> $setup_log 2>&1
 
 } | progress
 
@@ -620,7 +619,9 @@ if [[ -n $ALLOW_ROLE && -n $ALLOW_CIDR ]]; then
 	IP=$ALLOW_CIDR so-allow -$ALLOW_ROLE >> $setup_log 2>&1
 fi
 
-if [[ $success != 0 || $SO_ERROR == 1 ]]; then whiptail_setup_failed
+if [[ $success != 0 || $SO_ERROR == 1 ]]; then 
+	SKIP_REBOOT=1
+	whiptail_setup_failed
 else 
 	whiptail_setup_complete
 	if [[ $THEHIVE == 1 ]]; then check_hive_init; fi