diff --git a/benchmark/qps_test.py b/benchmark/qps_test.py
index 57a6c3e..367a5df 100644
--- a/benchmark/qps_test.py
+++ b/benchmark/qps_test.py
@@ -66,7 +66,7 @@ if args.num and args.pipe:
 
 if args.triton:
     client = triton_client.Client(url='localhost:8000')
-    out2 = client(data)[0][0][0]
+    out2 = client(data)[0][0]
     print('Triton: OK')
 
     if numpy.allclose(out1, out2, atol=args.atol):