From e3a25f1b11a811bb1b03198c7d2212d16bf0f2b2 Mon Sep 17 00:00:00 2001
From: Yousef Rafat <81116377+yousef-rafat@users.noreply.github.com>
Date: Thu, 14 May 2026 15:00:05 +0300
Subject: [PATCH] ..

---
 comfy_extras/nodes_trellis2.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/comfy_extras/nodes_trellis2.py b/comfy_extras/nodes_trellis2.py
index e4b9da53d..aa2f7b3ee 100644
--- a/comfy_extras/nodes_trellis2.py
+++ b/comfy_extras/nodes_trellis2.py
@@ -1841,12 +1841,12 @@ class Trellis2Extension(ComfyExtension):
     async def get_node_list(self) -> list[type[IO.ComfyNode]]:
         return [
             Trellis2Conditioning,
-            EmptyShapeLatentTrellis2,
-            EmptyStructureLatentTrellis2,
-            EmptyTextureLatentTrellis2,
-            VaeDecodeTrellis2Texture,
-            VaeDecodeTrellis2Shape,
-            VaeDecodeTrellis2Structure,
+            EmptyTrellis2ShapeLatent,
+            EmptyTrellis2LatentStructure,
+            EmptyTrellis2LatentTexture,
+            VaeDecodeTextureTrellis,
+            VaeDecodeShapeTrellis,
+            VaeDecodeStructureTrellis2,
             Trellis2UpsampleCascade,
             PostProcessMesh
         ]