diff --git a/setup/so-functions b/setup/so-functions
index 4f973d147..efa6c800f 100755
--- a/setup/so-functions
+++ b/setup/so-functions
@@ -101,8 +101,10 @@ desktop_salt_local() {
 		Press the Enter key to reboot.
 	EOM
 
-	whiptail --title "$whiptail_title" --msgbox "$message" 12 75
-	reboot
+	if [[ -z "$TESTING" ]]; then
+		whiptail --title "$whiptail_title" --msgbox "$message" 12 75
+		reboot
+	fi
 	exit 0 
 
 }