diff --git a/flake.nix b/flake.nix
index 708651b..676d2e1 100644
--- a/flake.nix
+++ b/flake.nix
@@ -52,7 +52,7 @@
         shen = lib.nixosSystem {
           inherit system;
           pkgs = pkgsForSystem;
-          specialArgs = { inherit unstable nixified-ai; };
+          specialArgs = { inherit unstable; };
           modules = [
             ./system/shen/configuration.nix
             simple-nixos-mailserver.nixosModule
diff --git a/modules/comfyui.nix b/modules/comfyui.nix
index dc2831e..ec6ea6c 100644
--- a/modules/comfyui.nix
+++ b/modules/comfyui.nix
@@ -1,4 +1,4 @@
-{ config, lib, pkgs, unstable, nixified-ai, ... }:
+{ config, lib, pkgs, unstable, ... }:
 
 let
   dn = "comfyui.tfcconnection.org";
@@ -12,7 +12,7 @@ in with lib;
       port = port;
       models = builtins.attrValues pkgs.nixified-ai.models;
       acceleration = "cuda";
-      customNodes = with nixified-ai.comfyui.pkgs; [
+      customNodes = with pkgs.nixified-ai.comfyui.pkgs; [
         comfyui-gguf
         comfyui-impact-pack