diff --git a/modules/comfyui.nix b/modules/comfyui.nix
index ec6ea6c..e67cec4 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.comfyui.pkgs; [
+      customNodes = with pkgs.nixified-ai.pkgs; [
         comfyui-gguf
         comfyui-impact-pack