diff --git a/flake.nix b/flake.nix
index fbff95f..99e4ce5 100644
--- a/flake.nix
+++ b/flake.nix
@@ -35,9 +35,6 @@
   }:
     flake-parts.lib.mkFlake {inherit inputs;} {
       systems = import systems;
-      imports = [
-        inputs.treefmt-nix.flakeModule
-        (import-tree ./parts)
-      ];
+      imports = [(import-tree ./parts)];
     };
 }
diff --git a/parts/treefmt.nix b/parts/treefmt.nix
index 968b0e3..ddec1e2 100644
--- a/parts/treefmt.nix
+++ b/parts/treefmt.nix
@@ -1,4 +1,6 @@
-_: {
+{inputs, ...}: {
+  imports = [inputs.treefmt-nix.flakeModule];
+
   perSystem = _: {
     treefmt = {
       programs = {