diff --git a/comfyui_models.yaml b/comfyui_models.yaml
index 7045e26..72c39dd 100644
--- a/comfyui_models.yaml
+++ b/comfyui_models.yaml
@@ -126,7 +126,7 @@ model_categories:
       size_gb: 3
       essential: false
       category: audio
-      type: checkpoints
+      type: audio_models
       format: fp32
       vram_gb: 4
       duration_seconds: 30
@@ -137,7 +137,7 @@ model_categories:
       size_gb: 11
       essential: true
       category: audio
-      type: checkpoints
+      type: audio_models
       format: fp32
       vram_gb: 8
       duration_seconds: 30
@@ -148,7 +148,7 @@ model_categories:
       size_gb: 22
       essential: false
       category: audio
-      type: checkpoints
+      type: audio_models
       format: fp32
       vram_gb: 16
       duration_seconds: 30