diff --git a/engines/python/setup/djl_python/fastertransformer.py b/engines/python/setup/djl_python/fastertransformer.py
index 35fffa29e4b..d08d3fe1fb5 100644
--- a/engines/python/setup/djl_python/fastertransformer.py
+++ b/engines/python/setup/djl_python/fastertransformer.py
@@ -115,7 +115,7 @@ def inference(self, inputs: Input):
                     batch_size=len(input_text),
                     output_len=output_len,
                     beam_width=beam_width,
-                    **parameters)[0]
+                    **parameters)
             result = [{"generated_text": s} for s in result]
             outputs = Output().add(result)
         except Exception as e: