diff --git a/flake.nix b/flake.nix
index bc8492c..f906bff 100644
--- a/flake.nix
+++ b/flake.nix
@@ -28,33 +28,33 @@
         "x86_64-darwin"
       ];
 
-      perSystem =
-        {
-          system,
-          pkgs,
-          self',
-          lib,
-          ...
-        }:
-        let
-          nixvim' = nixvim.legacyPackages.${system};
-          nvim = nixvim'.makeNixvimWithModule {
-            inherit pkgs;
-            module = ./config;
+      perSystem = {
+        system,
+        pkgs,
+        self',
+        lib,
+        ...
+      }: let
+        nixvimLib = nixvim.lib.${system};
+        nixvim' = nixvim.legacyPackages.${system};
+        nixvimModule = {
+          inherit pkgs;
+          module = import ./config; # import the module directly
+          # You can use `extraSpecialArgs` to pass additional arguments to your module files
+          extraSpecialArgs = {
+            # inherit (inputs) foo;
           };
-        in
-        {
-          checks = {
-            default = pkgs.nixvimLib.check.mkTestDerivationFromNvim {
-              inherit nvim;
-              name = "A nixvim configuration";
-            };
-            pre-commit-check = pre-commit-hooks.lib.${system}.run {
-              src = ./.;
-              hooks = {
-                statix.enable = true;
-                nixfmt-rfc-style.enable = true;
-              };
+        };
+        nvim = nixvim'.makeNixvimWithModule nixvimModule;
+      in
+      {
+        checks = {
+          default = nixvimLib.check.mkTestDerivationFromNixvimModule nixvimModule;
+          pre-commit-check = pre-commit-hooks.lib.${system}.run {
+            src = ./.;
+            hooks = {
+              statix.enable = true;
+              nixfmt-rfc-style.enable = true;
             };
           };