diff --git a/setup.sh b/setup.sh
index 275f9fa..2222cb5 100755
--- a/setup.sh
+++ b/setup.sh
@@ -455,7 +455,8 @@ main() {
         git clone "$GIT_REPO" "$DOTFILES_DIR" || die "Failed to clone dotfiles repository"
     fi
 
-    if [ "$1" = "--continue" ]; then
+    CONTINUE=$1
+    if [ "$CONTINUE" = "--continue" ]; then
         log_info "Continuing setup..."
     else
         warning_prompt
@@ -463,7 +464,7 @@ main() {
         check_selinux
         install_nix
     fi
-    
+
     install_home_manager
     setup_symlinks