diff --git a/templates/simple/flake.nix b/templates/simple/flake.nix
index 8f95a06c..69b600e1 100644
--- a/templates/simple/flake.nix
+++ b/templates/simple/flake.nix
@@ -11,9 +11,7 @@
     nixvim,
     flake-parts,
     ...
-  } @ inputs: let
-    config = import ./config; # import the module directly
-  in
+  } @ inputs:
     flake-parts.lib.mkFlake {inherit inputs;} {
       systems = [
         "x86_64-linux"
@@ -31,7 +29,7 @@
         nixvim' = nixvim.legacyPackages.${system};
         nixvimModule = {
           inherit pkgs;
-          module = config;
+          module = import ./config; # import the module directly
           # You can use `extraSpecialArgs` to pass additional arguments to your module files
           extraSpecialArgs = {
             # inherit (inputs) foo;