diff --git a/modules/comfyui.nix b/modules/comfyui.nix
index e67cec4..149a6ea 100644
--- a/modules/comfyui.nix
+++ b/modules/comfyui.nix
@@ -12,7 +12,7 @@ in with lib;
       port = port;
       models = builtins.attrValues pkgs.nixified-ai.models;
       acceleration = "cuda";
-      customNodes = with pkgs.nixified-ai.pkgs; [
+      customNodes = with comfyui.pkgs; [
         comfyui-gguf
         comfyui-impact-pack
 
diff --git a/system/shen/configuration.nix b/system/shen/configuration.nix
index d821b41..a551ae2 100644
--- a/system/shen/configuration.nix
+++ b/system/shen/configuration.nix
@@ -20,7 +20,7 @@ in
       ../../modules/mail.nix
       ../../modules/vaultwarden.nix
       ../../modules/nocodb.nix
-      ../../modules/comfyui.nix
+      # ../../modules/comfyui.nix
       # ../../pkgs/server.nix
       # ../../pkgs/nextcloud.nix
     ];