From edf95a72a91005f6be5402b17b145928e55d9256 Mon Sep 17 00:00:00 2001
From: jeff <113397187+cyberhorsey@users.noreply.github.com>
Date: Tue, 29 Aug 2023 20:10:48 -0700
Subject: [PATCH] fix(prover): add to wait group in prover (#373)

---
 prover/prover.go | 5 +++++
 1 file changed, 5 insertions(+)

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():