diff --git a/flake-parts/packages.nix b/flake-parts/packages.nix
index 339712cbc..45e8bcb48 100644
--- a/flake-parts/packages.nix
+++ b/flake-parts/packages.nix
@@ -7,6 +7,7 @@ let
       self.nixosModules.hidden-ssh-announce
       self.inputs.nixos-generators.nixosModules.all-formats
       self.inputs.disko.nixosModules.disko
+      ({ config, ... }: { system.stateVersion = config.system.nixos.version; })
     ];
   };
 in