diff --git a/modules/comfyui.nix b/modules/comfyui.nix
index 149a6ea..90acf9d 100644
--- a/modules/comfyui.nix
+++ b/modules/comfyui.nix
@@ -1,4 +1,4 @@
-{ config, lib, pkgs, unstable, ... }:
+{ config, lib, pkgs, unstable, comfyui, ... }:
 
 let
   dn = "comfyui.tfcconnection.org";