fix wg

0xPolygon · Nov 18, 2023 · 420f8d0 · 420f8d0
1 parent 9e2a8ce
commit 420f8d0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/e2e/datacommittee_test.go b/test/e2e/datacommittee_test.go
@@ -141,6 +141,7 @@ func TestDataCommittee(t *testing.T) {
 	// Start DAC nodes & DBs
 	wg := sync.WaitGroup{}
 	for _, m := range membs[1:] { // note starting all but first
+		wg.Add(1)
 		startDACMember(t, m, &wg)
 	}
 	wg.Wait()
@@ -270,7 +271,11 @@ func createKeyStore(pk *ecdsa.PrivateKey, outputDir, password string) error {
 }
 
 func startDACMember(t *testing.T, m member, wg *sync.WaitGroup) {
-	defer func() { wg.Done() }()
+	defer func() {
+		if wg != nil {
+			wg.Done()
+		}
+	}()
 	dacNodeConfig := config.Config{
 		L1: config.L1Config{
 			WsURL:                "ws://l1:8546",