diff --git a/comfyui_models.yaml b/comfyui_models.yaml
index 78ab3b2..341d136 100644
--- a/comfyui_models.yaml
+++ b/comfyui_models.yaml
@@ -32,7 +32,7 @@ model_categories:
       size_gb: 23
       essential: true
       category: image
-      type: diffusion_models
+      type: unet
       format: fp16
       vram_gb: 23
       notes: Industry-leading image generation quality
@@ -45,7 +45,7 @@ model_categories:
       size_gb: 23
       essential: false
       category: image
-      type: diffusion_models
+      type: unet
       format: fp16
       vram_gb: 23
       notes: Development version with enhanced features