diff --git a/comfy_extras/nodes_trellis2.py b/comfy_extras/nodes_trellis2.py
index 407ac105b..24ac373e2 100644
--- a/comfy_extras/nodes_trellis2.py
+++ b/comfy_extras/nodes_trellis2.py
@@ -1844,9 +1844,9 @@ class Trellis2Extension(ComfyExtension):
             EmptyShapeLatentTrellis2,
             EmptyStructureLatentTrellis2,
             EmptyTextureLatentTrellis2,
-            VaeDecodeTextureTrellis,
-            VaeDecodeShapeTrellis,
-            VaeDecodeStructureTrellis2,
+            VaeDecodeTrellis2Texture,
+            VaeDecodeTrellis2Shape,
+            VaeDecodeTrellis2Structure,
             Trellis2UpsampleCascade,
             PostProcessMesh
         ]