diff --git a/main.sh b/main.sh
index 0d59f4b42..302ac3024 100644
--- a/main.sh
+++ b/main.sh
@@ -141,7 +141,7 @@ if [[ $KERNEL_ONLY != yes && -z $RELEASE ]]; then
 	[[ -z $RELEASE ]] && exit_with_error "No release selected"
 fi
 
-if [[ $KERNEL_ONLY != yes && -n $RELEASE ]]; then
+if [[ $KERNEL_ONLY != yes && -z $BUILD_DESKTOP ]]; then
 	options=()
 	options+=("no" "Image with console interface")
 	options+=("yes" "Image with desktop environment")