diff --git a/nixos/nixified.ai.nix b/nixos/nixified.ai.nix
index 411b1e42..27031fd7 100644
--- a/nixos/nixified.ai.nix
+++ b/nixos/nixified.ai.nix
@@ -1,16 +1,11 @@
 { inputs, config, specialArgs, ... }:
 specialArgs.makeEnable config "myModules.nixified-ai" false {
-  # imports = [
-  #   inputs.nixified-ai.nixosModules.invokeai
-  # ];
+   imports = [
+     inputs.nixified-ai.nixosModules.comfyui
+   ];
 
-  # environment.systemPackages = [
-  #   inputs.nixified-ai.packages.${config.nixpkgs.system}.invokeai-nvidia
-  # ];
-
-  # services.invokeai = {
-  #   enable = false;
-  #   host = "0.0.0.0";
-  #   package = inputs.nixified-ai.packages.${config.nixpkgs.system}.invokeai-nvidia;
-  # };
+   services.comfyui = {
+     enable = true;
+     host = "0.0.0.0";
+   };
 }