diff --git a/flake.nix b/flake.nix
index 1e72e80..3e48b34 100644
--- a/flake.nix
+++ b/flake.nix
@@ -99,23 +99,23 @@
                 configFile = tomlFormat.generate "llm-shell-hint-config" cfg.settings;
               in
               lib.mkIf cfg.enable {
-                home.packages = [ self.packages."${system}".llm-shell-hint ];
+                home.packages = [ self.packages."${system}".default ];
 
                 programs.bash.initExtra = mkIf cfg.enableBashIntegration ''
                   ${
-                    lib.getExe self.packages."${system}".llm-shell-hint
+                    lib.getExe self.packages."${system}".default
                   } init bash --config-file "${configFile}" | source
                 '';
 
                 programs.zsh.initExtra = mkIf cfg.enableZshIntegration ''
                   ${
-                    lib.getExe self.packages."${system}".llm-shell-hint
+                    lib.getExe self.packages."${system}".default
                   } init zsh --config-file "${configFile}" | source
                 '';
 
                 programs.fish.interactiveShellInit = mkIf cfg.enableFishIntegration ''
                   ${
-                    lib.getExe self.packages."${system}".llm-shell-hint
+                    lib.getExe self.packages."${system}".default
                   } init fish --config-file "${configFile}" | source
                 '';
               };