diff --git a/flake-parts/checks.nix b/flake-parts/checks.nix
index 980b2f6..34c06d0 100644
--- a/flake-parts/checks.nix
+++ b/flake-parts/checks.nix
@@ -4,7 +4,10 @@
     { inputs', pkgs, ... }:
     {
       checks = {
-        formatting = (inputs.treefmt.lib.evalModule pkgs "${self}/treefmt.nix").config.build.check self;
+        formatting =
+          "${self}/treefmt.nix"
+          |> inputs.treefmt.lib.evalModule pkgs
+          |> (formatter: formatter.config.build.check self);
 
         statix = pkgs.runCommand "statix" { buildInputs = [ inputs'.statix.packages.statix ]; } ''
           statix check ${self}
diff --git a/flake-parts/formatter.nix b/flake-parts/formatter.nix
index e8ac02a..cb0703e 100644
--- a/flake-parts/formatter.nix
+++ b/flake-parts/formatter.nix
@@ -3,6 +3,9 @@
   perSystem =
     { pkgs, ... }:
     {
-      formatter = (inputs.treefmt.lib.evalModule pkgs "${self}/treefmt.nix").config.build.wrapper;
+      formatter =
+        "${self}/treefmt.nix"
+        |> inputs.treefmt.lib.evalModule pkgs
+        |> (formatter: formatter.config.build.wrapper);
     };
 }