diff --git a/flake.nix b/flake.nix
index ba03ad5..e144681 100644
--- a/flake.nix
+++ b/flake.nix
@@ -1,3 +1,4 @@
+# generation: 
 # flake for blakes nixos config
 # define new devices in outputs
 
diff --git a/rebuild.sh b/rebuild.sh
index 1197209..0f009dc 100755
--- a/rebuild.sh
+++ b/rebuild.sh
@@ -6,6 +6,7 @@ pushd ~/.nix
 git diff -U0 *.nix
 
 # add generation comment to flake.nix
+gen=$(nixos-rebuild list-generations | grep current | awk '{print $1}')
 if grep -q '^# generation:' flake.nix; then
     sed -i "1s/^# generation:.*/# generation: $gen/" flake.nix
 else