diff --git a/.forgejo/workflows/check.yml b/.forgejo/workflows/check.yml
index 0ab15443..34794508 100644
--- a/.forgejo/workflows/check.yml
+++ b/.forgejo/workflows/check.yml
@@ -10,7 +10,7 @@ jobs:
 
       - name: Check formatting
         run: |
-          nix --accept-flake-config --access-tokens '' develop --command treefmt --fail-on-change
+          nix --accept-flake-config --access-tokens '' develop --command treefmt --ci
 
       - name: Run flake checks
         run: |
diff --git a/flake.nix b/flake.nix
index a417b498..ce29df8a 100644
--- a/flake.nix
+++ b/flake.nix
@@ -107,7 +107,7 @@
               nvfetcher
               shellcheck
               shfmt
-              treefmt
+              inputs.unstable.legacyPackages.${system}.treefmt2
               nixos-generators
               inputs.unstable.legacyPackages.${system}.opentofu
               terraform-backend-git