diff --git a/modules/nixos/services/comfyui/default.nix b/modules/nixos/services/comfyui/default.nix
index 5b7f327..92e7609 100644
--- a/modules/nixos/services/comfyui/default.nix
+++ b/modules/nixos/services/comfyui/default.nix
@@ -1,6 +1,9 @@
-{ pkgs, ... }:
 {
-  environment.systemPackages = [
-    pkgs.comfyuiPackages
+  pkgs,
+  ...
+}:
+{
+  environment.comfyuiPackages = [
+    pkgs.comfyui
   ];
 }