diff --git a/comfy/ldm/hunyuan3d/model/moe.py b/comfy/ldm/hunyuan3d/model/moe.py
index f6ea7db3c..93becdfbb 100644
--- a/comfy/ldm/hunyuan3d/model/moe.py
+++ b/comfy/ldm/hunyuan3d/model/moe.py
@@ -152,6 +152,8 @@ class MoEBlock(nn.Module):
 
         y = y + self.shared_experts(identity)
 
+        return y
+
     @torch.no_grad()
     def moe_infer(self, x, flat_expert_indices, flat_expert_weights):
         
@@ -200,4 +202,4 @@ def test_moe():
     print(timing)
 
 if __name__ == "__main__":
-    test_moe()
\ No newline at end of file
+    test_moe()