Fix error: related to samza batching