diff --git a/pkgs/by-name/ni/nixfmt/update.sh b/pkgs/by-name/ni/nixfmt/update.sh
index 3a44a0c6279a..f6890b115ef2 100755
--- a/pkgs/by-name/ni/nixfmt/update.sh
+++ b/pkgs/by-name/ni/nixfmt/update.sh
@@ -1,5 +1,5 @@
 #!/usr/bin/env nix-shell
-#!nix-shell -i bash -p cabal2nix curl jq
+#!nix-shell --pure -i bash -p cabal2nix curl cacert jq nix
 #
 # This script will update the nixfmt derivation to the latest version using
 # cabal2nix.
@@ -24,4 +24,7 @@ cabal2nix --jailbreak \
   "https://github.com/nixos/nixfmt/archive/${release_tag}.tar.gz" \
   >> "$derivation_file"
 
+nix-shell "$script_dir/../../../../" \
+  --run "treefmt --no-cache $derivation_file"
+
 echo "Finished."