diff --git a/templates/simple/config/default.nix b/templates/simple/config/default.nix
index fb318aba7e..bf671e2004 100644
--- a/templates/simple/config/default.nix
+++ b/templates/simple/config/default.nix
@@ -1,4 +1,7 @@
+{ inputs, system, ... }:
 {
   # Import all your configuration modules here
   imports = [ ./bufferline.nix ];
+
+  nixpkgs.pkgs = import inputs.nixpkgs { inherit system; };
 }
diff --git a/templates/simple/flake.nix b/templates/simple/flake.nix
index 8e80918683..572bf705f8 100644
--- a/templates/simple/flake.nix
+++ b/templates/simple/flake.nix
@@ -23,11 +23,10 @@
           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;
+              inherit inputs system;
             };
           };
           nvim = nixvim'.makeNixvimWithModule nixvimModule;