diff --git a/prover/prover.go b/prover/prover.go
index ea3738659..e54a8dce2 100644
--- 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():