diff --git a/README.md b/README.md
index f06423d..2fc41c6 100644
--- a/README.md
+++ b/README.md
@@ -198,7 +198,7 @@ Add the module to your NixOS configuration:
 | Option | Type | Default | Description |
 |--------|------|---------|-------------|
 | `enable` | bool | `false` | Enable the listener service |
-| `package` | package | `pkgs.homelab-deploy` | Package to use |
+| `package` | package | from flake | Package to use |
 | `hostname` | string | `config.networking.hostName` | Hostname for subject templates |
 | `tier` | enum | required | `"test"` or `"prod"` |
 | `role` | string | `null` | Role for role-based targeting |
diff --git a/flake.nix b/flake.nix
index f8542bf..1c37b55 100644
--- a/flake.nix
+++ b/flake.nix
@@ -49,7 +49,7 @@
           };
         });
 
-      nixosModules.default = import ./nixos/module.nix;
+      nixosModules.default = import ./nixos/module.nix { inherit self; };
       nixosModules.homelab-deploy = self.nixosModules.default;
     };
 }
diff --git a/nixos/module.nix b/nixos/module.nix
index 9022616..03c2843 100644
--- a/nixos/module.nix
+++ b/nixos/module.nix
@@ -1,3 +1,4 @@
+{ self }:
 { config, lib, pkgs, ... }:
 
 let
@@ -21,7 +22,11 @@ in
   options.services.homelab-deploy.listener = {
     enable = lib.mkEnableOption "homelab-deploy listener service";
 
-    package = lib.mkPackageOption pkgs "homelab-deploy" { };
+    package = lib.mkOption {
+      type = lib.types.package;
+      default = self.packages.${pkgs.system}.homelab-deploy;
+      description = "The homelab-deploy package to use";
+    };
 
     hostname = lib.mkOption {
       type = lib.types.str;