diff --git a/wait_updates.go b/wait_updates.go
index 1ff92be..958b452 100644
--- a/wait_updates.go
+++ b/wait_updates.go
@@ -12,6 +12,7 @@ func (b *Bot) waitUpdates(ctx context.Context, wg *sync.WaitGroup) {
 	for {
 		select {
 		case <-ctx.Done():
+			return
 		case upd := <-b.updates:
 			b.ProcessUpdate(ctx, upd)
 		}