diff --git a/tests/default.nix b/tests/default.nix
index dae54920..4d7a18c9 100644
--- a/tests/default.nix
+++ b/tests/default.nix
@@ -10,9 +10,13 @@ let
   inherit (test-derivation) mkTestDerivationFromNixvimModule;
 
   moduleToTest =
-    name: module:
+    file: name: module:
     mkTestDerivationFromNixvimModule {
-      inherit name module;
+      inherit name;
+      module = {
+        _file = file;
+        imports = [ module ];
+      };
       pkgs = pkgsUnfree;
     };
 
@@ -39,10 +43,17 @@ let
 in
 # We attempt to build & execute all configurations
 lib.pipe (testFiles ++ [ exampleFiles ]) [
-  (builtins.map (file: {
-    inherit (file) name;
-    path = pkgs.linkFarm file.name (builtins.mapAttrs moduleToTest file.cases);
-  }))
+  (builtins.map (
+    {
+      name,
+      file,
+      cases,
+    }:
+    {
+      inherit name;
+      path = pkgs.linkFarm name (builtins.mapAttrs (moduleToTest file) cases);
+    }
+  ))
   (helpers.groupListBySize 10)
   (lib.imap1 (
     i: group: rec {