diff --git a/flake.nix b/flake.nix
index cc212c740..5f985ddd3 100644
--- a/flake.nix
+++ b/flake.nix
@@ -52,6 +52,7 @@
       ...
     }: {
       imports = [./nix/home-module.nix];
+      programs.noctalia-shell.package = lib.mkDefault self.packages.${pkgs.system}.default;
       programs.noctalia-shell.app2unit.package =
         lib.mkDefault
         nixpkgs.legacyPackages.${pkgs.system}.app2unit;
diff --git a/nix/home-module.nix b/nix/home-module.nix
index 084109111..713c8f65c 100644
--- a/nix/home-module.nix
+++ b/nix/home-module.nix
@@ -16,6 +16,11 @@ in {
   options.programs.noctalia-shell = {
     enable = lib.mkEnableOption "Noctalia shell configuration";
 
+    package = lib.mkOption {
+      type = lib.types.nullOr lib.types.package;
+      description = "The noctalia-shell package to use";
+    };
+
     settings = lib.mkOption {
       type = with lib.types;
         nullOr (oneOf [
@@ -98,7 +103,8 @@ in {
     useApp2Unit = cfg.settings.appLauncher.useApp2Unit or false;
   in
     lib.mkIf cfg.enable {
-      home.packages = lib.optional useApp2Unit cfg.app2unit.package;
+      home.packages = lib.optional useApp2Unit cfg.app2unit.package
+        ++ lib.optional (cfg.package != null) cfg.package;
 
       xdg.configFile = {
         "noctalia/settings.json" = {