diff --git a/raspi-config b/raspi-config
index 48dbef91..5df5d1f0 100755
--- a/raspi-config
+++ b/raspi-config
@@ -2814,7 +2814,7 @@ do_overlayfs() {
       whiptail --yesno "Would you like the boot partition to be write-protected?" $DEFAULT 20 60 2
       RET=$?
     else
-      RET=$1
+      RET=${2:-$1}
     fi
     if [ $RET -eq $CURRENT ]; then
       if [ $RET -eq 0 ]; then