diff --git a/energon/server/batch_manager.py b/energon/server/batch_manager.py
index a4a55e7..1a7b2a5 100644
--- a/energon/server/batch_manager.py
+++ b/energon/server/batch_manager.py
@@ -241,7 +241,7 @@ def processing_batch(self):
                 else:
                     input_ids = input_text
                 output = self.engine.run(input_ids)
-                self.pool.submit(self.publish_result, output, target_batch, start_time)
+                self.pool.submit(self.publish_result, output, target_batch)
                 # self.publish_result(output, target_batch, start_time)
                 # pub_thread = threading.Thread(target=self.publish_result, args=(output, target_batch, start_time))
                 # pub_thread.start()