diff --git a/setup/so-setup b/setup/so-setup
index 17c62af81..60296d2f3 100755
--- a/setup/so-setup
+++ b/setup/so-setup
@@ -91,7 +91,7 @@ fi
 # if packages are updated and the box isn't rebooted
 if [[ $is_debian ]]; then
 	update_packages
-	if [[ -f "/var/run/reboot-required" ]] && [ -z "$TESTING" ]; then
+	if [[ -f "/var/run/reboot-required" ]] && [ -z "$test_profile" ]; then
 		whiptail_debian_reboot_required
 		reboot
 	fi