diff --git a/deploy.sh b/deploy.sh
index c9813cf..78ff99f 100755
--- a/deploy.sh
+++ b/deploy.sh
@@ -1,2 +1,7 @@
 #!/bin/sh
-nixos-rebuild boot --flake .#muffin --target-host root@server-public --build-host root@server-public --verbose
+ARG="$*"
+if [ "$ARG" = "" ]; then
+  ARG="boot"
+fi
+
+nixos-rebuild "$ARG" --flake .#muffin --target-host root@server-public --build-host root@server-public --verbose