diff --git a/ants_test.go b/ants_test.go
index 1193aff..5dfe95d 100644
--- a/ants_test.go
+++ b/ants_test.go
@@ -427,24 +427,25 @@ func TestMaxBlockingSubmit(t *testing.T) {
 
 func TestNonblockingSubmitWithFunc(t *testing.T) {
 	poolSize := 10
-	ch1 := make(chan struct{})
+	var wg sync.WaitGroup
 	p, err := NewPoolWithFunc(poolSize, func(i interface{}) {
 		longRunningPoolFunc(i)
-		close(ch1)
+		wg.Done()
 	}, WithNonblocking(true))
 	assert.NoError(t, err, "create TimingPool failed: %v", err)
 	defer p.Release()
-	for i := 0; i < poolSize-1; i++ {
-		assert.NoError(t, p.Invoke(nil), "nonblocking submit when pool is not full shouldn't return error")
-	}
 	ch := make(chan struct{})
+	wg.Add(poolSize)
+	for i := 0; i < poolSize-1; i++ {
+		assert.NoError(t, p.Invoke(ch), "nonblocking submit when pool is not full shouldn't return error")
+	}
 	// p is full now.
 	assert.NoError(t, p.Invoke(ch), "nonblocking submit when pool is not full shouldn't return error")
 	assert.EqualError(t, p.Invoke(nil), ErrPoolOverload.Error(),
 		"nonblocking submit when pool is full should get an ErrPoolOverload")
 	// interrupt f to get an available worker
 	close(ch)
-	<-ch1
+	wg.Wait()
 	assert.NoError(t, p.Invoke(nil), "nonblocking submit when pool is not full shouldn't return error")
 }