diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py
index bf8cd956..3f2ffa8b 100644
--- a/api/onnx_web/server/api.py
+++ b/api/onnx_web/server/api.py
@@ -455,7 +455,7 @@ def chain(server: ServerContext, pool: DevicePoolExecutor):
         server,
         params,
         [],
-        output=output[0],
+        output=output,
         size=size,
         needs_device=device,
     )