diff --git a/vllm_omni/diffusion/models/ltx2/pipeline_ltx2.py b/vllm_omni/diffusion/models/ltx2/pipeline_ltx2.py
index ba95db235..4cc2de28b 100644
--- a/vllm_omni/diffusion/models/ltx2/pipeline_ltx2.py
+++ b/vllm_omni/diffusion/models/ltx2/pipeline_ltx2.py
@@ -190,7 +190,4 @@ class LTX2Pipeline(nn.Module):
             frames = output
         
         # Return as DiffusionOutput
-        return DiffusionOutput(
-            images=frames,
-            final_output_type="image",
-        )
+        return DiffusionOutput(output=frames)