diff --git a/flake-module.nix b/flake-module.nix
index 4b336437..2fb18d07 100644
--- a/flake-module.nix
+++ b/flake-module.nix
@@ -68,7 +68,7 @@ in
             rootSrc = self.outPath;
             package = lib.mkDefault pkgs.pre-commit;
             tools = import ./nix/call-tools.nix pkgs;
-            settings.treefmt.package = lib.mkIf (options?treefmt) (lib.mkDefault config.treefmt.build.wrapper);
+            hooks.treefmt.package = lib.mkIf (options?treefmt) (lib.mkOverride 900 config.treefmt.build.wrapper);
           };
           pre-commit.devShell = pkgs.mkShell {
             nativeBuildInputs = [ cfg.settings.package ];
diff --git a/nix/default.nix b/nix/default.nix
index 2e73c5f2..edbb0924 100644
--- a/nix/default.nix
+++ b/nix/default.nix
@@ -32,7 +32,6 @@ let
                 ../modules/all-modules.nix
                 {
                   inherit tools;
-                  hooks.treefmt.package = pkgs.treefmt;
                 }
               ];
               specialArgs = { inherit pkgs; };