diff --git a/rebuild.sh b/rebuild.sh
index b7d456c..96f3b81 100755
--- a/rebuild.sh
+++ b/rebuild.sh
@@ -4,15 +4,8 @@ pushd ~/.nix
 # nvim flake.nix
 # alejandra . &>/dev/null
 # git diff -U0 *.nix
-git diff -U0 $(find . -name '*.nix')
 
 # add generation comment to flake.nix
-gen=$(nixos-rebuild list-generations | grep current | awk '{print $1}')
-
-echo "nixos rebuilding..."
-sudo nixos-rebuild switch --flake ~/.nix#snowbelle &>.nixos-switch-log || (
- cat .nixos-switch-log)
-
 gen=$(nixos-rebuild list-generations | grep current)
 if sed -n '3p' flake.nix | grep -q '^# generation:'; then
     # replace the comment on line 3
@@ -22,5 +15,14 @@ else
     sed -i "3i# generation: $gen" flake.nix
 fi
 
+
+
+git diff -U0 $(find . -name '*.nix')
+
+
+echo "nixos rebuilding..."
+sudo nixos-rebuild switch --flake ~/.nix#snowbelle &>.nixos-switch-log || (
+ cat .nixos-switch-log | grep --color error && false)
+
 git commit -am "$gen"
 popd