diff --git a/justfile b/justfile
index 707fc77..68ad4b3 100644
--- a/justfile
+++ b/justfile
@@ -3,17 +3,9 @@ set quiet := true
 default:
     just --list --unsorted
 
-switch:
-    nh os switch .
-
-test:
-    nh os test .
-
-boot:
-    nh os boot .
-
-reboot: boot
-    reboot
+rebuild mode='switch':
+    nh os {{ if mode == 'reboot' { 'boot' } else { mode } }} .
+    {{ if mode == 'reboot' { 'reboot' } else { '' } }}
 
 update:
     nix flake update