fix(prover): add to wait group in prover (#373)

taikoxyz · Aug 30, 2023 · edf95a7 · edf95a7
1 parent e09fd97
commit edf95a7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/prover/prover.go b/prover/prover.go
@@ -245,6 +245,11 @@ func (p *Prover) Start() error {
 }
 
 func (p *Prover) watchCurrentCapacity() {
+	p.wg.Add(1)
+	defer func() {
+		p.wg.Done()
+	}()
+
 	for {
 		select {
 		case <-p.ctx.Done():