diff --git a/nodes.py b/nodes.py
index 6a80d8d..65fd380 100644
--- a/nodes.py
+++ b/nodes.py
@@ -92,7 +92,7 @@ class DownloadAndLoadMochiModel:
                 ),
                 "vae": (
                     [   
-                        "mochi_preview_vae_bf16.safetensors",
+                        "mochi_preview_vae_decoder_bf16.safetensors",
                     ],
                     {"tooltip": "Downloads from 'https://huggingface.co/Kijai/Mochi_preview_comfy' to 'models/vae/mochi'", },
                 ),