diff --git a/producer/processor.go b/producer/processor.go
index 7428386..32b6cb4 100644
--- a/producer/processor.go
+++ b/producer/processor.go
@@ -239,7 +239,7 @@ func AggregateProcessorChannel(
 	)
 
 	go func() {
-		<-aggHandler.Done()
+		<-mainHandler.Done()
 		aggHandler.GracefulWait()
 		close(tin)
 		close(fin)