diff --git a/server/lorax_server/server.py b/server/lorax_server/server.py
index 82ada7ff7..066120f3d 100644
--- a/server/lorax_server/server.py
+++ b/server/lorax_server/server.py
@@ -114,7 +114,7 @@ async def Decode(self, request, context):
             raise ValueError("All batches are empty")
 
         if len(batches) > 1:
-            batch = self.model.batch_type.concatenate(batches, self.model.tokenizer, self.model.tokenizers)
+            batch = self.model.batch_type.concatenate(batches)
         else:
             batch = batches[0]