diff --git a/modules/nixos/services/comfyui/default.nix b/modules/nixos/services/comfyui/default.nix
index 04d4794..90cb702 100755
--- a/modules/nixos/services/comfyui/default.nix
+++ b/modules/nixos/services/comfyui/default.nix
@@ -19,27 +19,22 @@ in
 
       package = mkOption {
         type = types.package;
-        default = pkgs.comfyuiPackages.comfyui;
+        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";
       };
-      # 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";
-      # };
 
       port = mkOption {
         type = types.port;
@@ -76,7 +71,6 @@ in
       serviceConfig = {
         Type = "simple";
         ExecStart = "${cfg.package}/bin/comfyui --port ${toString cfg.port} --listen ${cfg.host}";
-        ExecStartPost = "${pkgs.bash}/bin/bash -c 'sleep 5 && mkdir -p /var/lib/comfyui/custom_nodes'";
         Restart = "on-failure";
         User = "comfyui";
         Group = "comfyui";
diff --git a/systems/ceres/config/comfyui.nix b/systems/ceres/config/comfyui.nix
index 08f4cf4..4a8a209 100755
--- a/systems/ceres/config/comfyui.nix
+++ b/systems/ceres/config/comfyui.nix
@@ -22,21 +22,21 @@ in
       enable = true;
       openFirewall = true;
       host = localhost;
-      # package = pkgs.comfyuiPackages.comfyui.override {
-      #   extensions = with pkgs.comfyuiPackages.extensions; [
-      #     acly-inpaint
-      #     acly-tooling
-      #     kosinkadink-advanced-controlnet
-      #     kosinkadink-animatediff-evolved
-      #     kosinkadink-video-helper-suite
-      #     lev145-images-grid
-      #     ssitu-ultimate-sd-upscale
-      #   ];
-      #   commandLineArgs = [
-      #     "--preview-method"
-      #     "auto"
-      #   ];
-      # };
+      package = pkgs.comfyuiPackages.comfyui.override {
+        extensions = with pkgs.comfyuiPackages.extensions; [
+          acly-inpaint
+          acly-tooling
+          kosinkadink-advanced-controlnet
+          kosinkadink-animatediff-evolved
+          kosinkadink-video-helper-suite
+          lev145-images-grid
+          ssitu-ultimate-sd-upscale
+        ];
+        commandLineArgs = [
+          "--preview-method"
+          "auto"
+        ];
+      };
     };
     caddy = {
       virtualHosts = {
@@ -66,7 +66,6 @@ in
     ];
   };
   systemd.tmpfiles.rules = [
-    # "d /var/lib/${service.name}/custom_nodes 755 ${service.name} ${service.name} -"
     "Z ${service.paths.path0}/models 755 root root -"
     "Z ${service.sops.path0} 755 caddy caddy -"
   ];