Wait timeout + 1s before exit

1 files changed, 2 insertions, 2 deletions

diff --git a/main.go b/main.go
index b726dd1..e4d2027 100644
--- a/main.go
+++ b/main.go
@@ -202,7 +202,7 @@ func workAggregator(ctx context.Context, opts options.Options, answerCh chan qna
 				select {
 				case a := <-answerCh:
 					handleAnswer(a)
-				case <-time.After(opts.Timeout):
+				case <-time.After(opts.Timeout + time.Second):
 					return
 				}
 			}
@@ -260,7 +260,7 @@ func workGenerator(ctx context.Context, opts options.Options, fp *os.File, quest
 
 	select {
 	case <-ctx.Done():
-	case <-time.After(opts.Timeout):
+	case <-time.After(opts.Timeout + time.Second):
 	}
 	return nextLine, true
 }