diff --git a/flake.nix b/flake.nix
index 79d0c45..9d38251 100644
--- a/flake.nix
+++ b/flake.nix
@@ -103,7 +103,10 @@
         inherit lanzatool;
       };
 
-      nixosModules.lanzaboote = import ./nix/lanzaboote.nix;
+      nixosModules.lanzaboote = { pkgs, lib, ... }: {
+        imports = [ ./nix/lanzaboote.nix ];
+        boot.lanzaboote.package = lib.mkDefault self.packages.${pkgs.system}.lanzaboote;
+      };
 
       packages.x86_64-linux = {
         inherit initrd-stub lanzaboote lanzatool;