diff --git a/home/baseline/nushell/functions.nu b/home/baseline/nushell/functions.nu
index 96da615..9344952 100644
--- a/home/baseline/nushell/functions.nu
+++ b/home/baseline/nushell/functions.nu
@@ -12,15 +12,15 @@ def --env unlock [] {
     }
 }
 
-def rebuild [] {
+def rebuild [ $target: string = "switch" ] {
     if (uname | get operating-system) == "Darwin" {
-        sudo darwin-rebuild switch
+        sudo darwin-rebuild $target
     } else {
         let hostname = uname | get nodename
         let build = ^nom build --keep-going $"/etc/nixos#nixosConfigurations.($hostname).config.system.build.toplevel"
         if $env.LAST_EXIT_CODE == 0 {
             nvd diff /run/current-system result
-            run0 result/bin/switch-to-configuration switch
+            run0 result/bin/switch-to-configuration $target
         } else {
             print "Error during build"
         }