diff --git a/modules/nixos/services/comfyui/default.nix b/modules/nixos/services/comfyui/default.nix
index 802eb49..1679abd 100644
--- a/modules/nixos/services/comfyui/default.nix
+++ b/modules/nixos/services/comfyui/default.nix
@@ -9,42 +9,48 @@ let
   cfg = config.services.comfyui;
 in
 {
-  options.services.comfyui = with lib; {
-    enable = mkEnableOption "ComfyUI service";
+  imports = [ ];
 
-    package = mkOption {
-      type = types.package;
-      default = pkgs.comfyuiPackages.comfyui.override {
-        extensions = with pkgs.comfyuiPackages.extensions; [
-          acly-inpaint
-          acly-tooling
-          cubiq-ipadapter-plus
-          fannovel16-controlnet-aux
-        ];
-        commandLineArgs = [
-          "--preview-method"
-          "auto"
-        ];
+  options = {
+    services.comfyui = with lib; {
+      enable = mkEnableOption "ComfyUI service";
+
+      package = mkOption {
+        type = types.package;
+        default = pkgs.comfyuiPackages.comfyui.override {
+          extensions = with pkgs.comfyuiPackages.extensions; [
+            # Add desired extensions here
+            # Example extensions:
+            # acly-inpaint
+            # acly-tooling
+            # cubiq-ipadapter-plus
+            # fannovel16-controlnet-aux
+          ];
+          commandLineArgs = [
+            "--preview-method"
+            "auto"
+          ];
+        };
+        description = "The ComfyUI package to use";
       };
-      description = "The ComfyUI package to use";
-    };
 
-    port = mkOption {
-      type = types.port;
-      default = 8188;
-      description = "Port to listen on";
-    };
+      port = mkOption {
+        type = types.port;
+        default = 8188;
+        description = "Port to listen on";
+      };
 
-    host = mkOption {
-      type = types.str;
-      default = "0.0.0.0";
-      description = "Host to bind to";
-    };
+      host = mkOption {
+        type = types.str;
+        default = "127.0.0.1";
+        description = "Host to bind to";
+      };
 
-    openFirewall = mkOption {
-      type = types.bool;
-      default = false;
-      description = "Open ports in the firewall";
+      openFirewall = mkOption {
+        type = types.bool;
+        default = false;
+        description = "Open ports in the firewall";
+      };
     };
   };
 
@@ -79,19 +85,4 @@ in
 
     users.groups.comfyui = { };
   };
-  services.comfyui = {
-    enable = true;
-    package = pkgs.comfyuiPackages.comfyui.override {
-      extensions = with pkgs.comfyuiPackages.extensions; [
-        acly-inpaint
-        acly-tooling
-        cubiq-ipadapter-plus
-        fannovel16-controlnet-aux
-      ];
-      commandLineArgs = [
-        "--preview-method"
-        "auto"
-      ];
-    };
-  };
 }