diff --git a/modules/programs/zsh.nix b/modules/programs/zsh.nix
index 8b0f761..ddfee64 100644
--- a/modules/programs/zsh.nix
+++ b/modules/programs/zsh.nix
@@ -1,4 +1,4 @@
-{ inputs, ... }:
+{ self, inputs, ... }:
 let
   username = "john";
 in
@@ -50,16 +50,18 @@ in
       };
     };
   };
-  
-  flake.wrappers.zsh = { pkgs, wlib, ... }: {
-    imports = [wlib.wrapperModules.zsh];
-    extraPackages = with pkgs; [
-      btop
-      coreutils
-      curl
-      wget
-      yazi
-      zsh
-    ];
+
+  perSystem = { pkgs, ... }: {
+    packages.jsl-zsh = inputs.wrapper-modules.wrappers.zsh.wrap {
+      inherit pkgs;
+      extraPackages = with pkgs; [
+        btop
+        coreutils
+        curl
+        wget
+        yazi
+        zsh
+      ];
+    };
   };
 }
\ No newline at end of file