diff --git a/produce.go b/produce.go
index 9ff413a..3493c58 100644
--- a/produce.go
+++ b/produce.go
@@ -329,6 +329,7 @@ func batchRecords(wg *sync.WaitGroup, in chan message, out chan []message) {
 		select {
 		case m, ok := <-in:
 			if !ok {
+				send()
 				return
 			}