diff --git a/flake.lock b/flake.lock
index 5ab252b1b..e717d4f6b 100644
--- a/flake.lock
+++ b/flake.lock
@@ -131,11 +131,11 @@
         ]
       },
       "locked": {
-        "lastModified": 1724833132,
-        "narHash": "sha256-F4djBvyNRAXGusJiNYInqR6zIMI3rvlp6WiKwsRISos=",
+        "lastModified": 1725271338,
+        "narHash": "sha256-FuP8Im+wj4xxaUp1bffhgvJvXtUT6TapkMfRcjaIAK4=",
         "owner": "numtide",
         "repo": "treefmt-nix",
-        "rev": "3ffd842a5f50f435d3e603312eefa4790db46af5",
+        "rev": "23c2b0d953710939487ec878d70495d73b2c9bb3",
         "type": "github"
       },
       "original": {
diff --git a/formatter.nix b/formatter.nix
index 33f87e5ac..69cd47c33 100644
--- a/formatter.nix
+++ b/formatter.nix
@@ -46,15 +46,11 @@
       };
       treefmt.programs.mypy.directories =
         {
-          "pkgs/clan-cli" = {
-            extraPythonPackages = self'.packages.clan-cli.testDependencies;
-            modules = [ "clan_cli" ];
-          };
+          "pkgs/clan-cli".extraPythonPackages = self'.packages.clan-cli.testDependencies;
           "pkgs/clan-app" = {
             extraPythonPackages =
               (self'.packages.clan-app.externalTestDeps or [ ]) ++ self'.packages.clan-cli.testDependencies;
             extraPythonPaths = [ "../clan-cli" ];
-            modules = [ "clan_app" ];
           };
         }
         // (
@@ -64,7 +60,6 @@
                 extraPythonPackages =
                   self'.packages.clan-vm-manager.externalTestDeps ++ self'.packages.clan-cli.testDependencies;
                 extraPythonPaths = [ "../clan-cli" ];
-                modules = [ "clan_vm_manager" ];
               };
             }
           else
@@ -72,6 +67,5 @@
         );
       treefmt.programs.ruff.check = true;
       treefmt.programs.ruff.format = true;
-
     };
 }