diff --git a/ants_benchmark_test.go b/ants_benchmark_test.go
index 0c7728d..566c794 100644
--- a/ants_benchmark_test.go
+++ b/ants_benchmark_test.go
@@ -77,13 +77,14 @@ func BenchmarkGoroutineWithFunc(b *testing.B) {
 }
 
 func BenchmarkAntsPoolWithFunc(b *testing.B) {
-	defer ants.Release()
 	var wg sync.WaitGroup
 	p, _ := ants.NewPoolWithFunc(50000, func(i interface{}) error {
 		demoPoolFunc(i)
 		wg.Done()
 		return nil
 	})
+	p.Release()
+
 	for i := 0; i < b.N; i++ {
 		for j := 0; j < RunTimes; j++ {
 			wg.Add(1)
diff --git a/ants_test.go b/ants_test.go
index 0e1bc26..11059ec 100644
--- a/ants_test.go
+++ b/ants_test.go
@@ -33,6 +33,7 @@ import (
 var n = 1000000
 
 func TestDefaultPool(t *testing.T) {
+	defer ants.Release()
 	var wg sync.WaitGroup
 	for i := 0; i < n; i++ {
 		wg.Add(1)
@@ -54,7 +55,6 @@ func TestDefaultPool(t *testing.T) {
 }
 
 func TestNoPool(t *testing.T) {
-	defer ants.Release()
 	var wg sync.WaitGroup
 	for i := 0; i < n; i++ {
 		wg.Add(1)