diff --git a/deploy.sh b/deploy.sh
index c90a1c3..76a0e9a 100755
--- a/deploy.sh
+++ b/deploy.sh
@@ -1,8 +1,8 @@
 #!/bin/sh
 
-ARG="$@"
-if [ "$ARG" == "" ]; then
+ARG="$*"
+if [ "$ARG" = "" ]; then
   ARG="boot"
 fi
 
-nixos-rebuild $ARG --flake . --use-remote-sudo
+nixos-rebuild "$ARG" --flake . --use-remote-sudo