diff --git a/justfile b/justfile
index a1d0bcb..c2373ce 100644
--- a/justfile
+++ b/justfile
@@ -1,5 +1,7 @@
 set export
 
+alias u := update
+
 config := "machines/$(hostname)/configuration.nix"
 arch := `uname -s`
 
@@ -9,7 +11,7 @@ rebuild:
   $cmd switch --flake . -I nixos-config=$config
 
 rollback:
-  $cmd switch --rollback -I nixos-config=$config
+  $cmd switch --rollback --flake .
 
 update:
   @./nix-flake-update.sh