diff --git a/modules/features/flameshot.nix b/modules/features/flameshot.nix
index 538d331..0db5004 100644
--- a/modules/features/flameshot.nix
+++ b/modules/features/flameshot.nix
@@ -1,5 +1,13 @@
 {
-  flake.nixosModules.flameshot = {
+  flake.nixosModules.flameshot = { pkgs, ... }: {
+    environment.systemPackages = with pkgs; [
+      flameshot
+      grim
+      slurp
+      satty
+      wl-copy
+    ];
+
     services.flameshot = {
       enable = true;
       General = {
@@ -10,12 +18,4 @@
     };
   };
 
-
-  environment.systemPackages = [
-    flameshot
-    grim
-    slurp
-    satty
-    wl-copy
-  ];
 }