diff --git a/benchmarks/dnn/layers/conv2d_test.go b/benchmarks/dnn/layers/conv2d_test.go
index 8191126b..14f9f88b 100644
--- a/benchmarks/dnn/layers/conv2d_test.go
+++ b/benchmarks/dnn/layers/conv2d_test.go
@@ -1,7 +1,7 @@
 package layers
 
 import (
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 	"github.com/sarchlab/mgpusim/v3/benchmarks/dnn/tensor"
 )
diff --git a/benchmarks/dnn/layers/fullyconnectedlayer_test.go b/benchmarks/dnn/layers/fullyconnectedlayer_test.go
index 99debd20..60fe8347 100644
--- a/benchmarks/dnn/layers/fullyconnectedlayer_test.go
+++ b/benchmarks/dnn/layers/fullyconnectedlayer_test.go
@@ -1,7 +1,7 @@
 package layers
 
 import (
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 	"github.com/sarchlab/mgpusim/v3/benchmarks/dnn/tensor"
 )
diff --git a/benchmarks/dnn/layers/layers_suite_test.go b/benchmarks/dnn/layers/layers_suite_test.go
index a89d53bb..9459e0db 100644
--- a/benchmarks/dnn/layers/layers_suite_test.go
+++ b/benchmarks/dnn/layers/layers_suite_test.go
@@ -3,7 +3,7 @@ package layers_test
 import (
 	"testing"
 
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 )
 
diff --git a/benchmarks/dnn/layers/relulayer_test.go b/benchmarks/dnn/layers/relulayer_test.go
index 38db09b7..532b343e 100644
--- a/benchmarks/dnn/layers/relulayer_test.go
+++ b/benchmarks/dnn/layers/relulayer_test.go
@@ -1,7 +1,7 @@
 package layers
 
 import (
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 	"github.com/sarchlab/mgpusim/v3/benchmarks/dnn/tensor"
 )
diff --git a/benchmarks/dnn/tensor/operator_test.go b/benchmarks/dnn/tensor/operator_test.go
index 75bfac0c..d09a9977 100644
--- a/benchmarks/dnn/tensor/operator_test.go
+++ b/benchmarks/dnn/tensor/operator_test.go
@@ -1,7 +1,7 @@
 package tensor
 
 import (
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 )
 
diff --git a/benchmarks/dnn/tensor/tensor_suite_test.go b/benchmarks/dnn/tensor/tensor_suite_test.go
index d12e2a03..f70d2171 100644
--- a/benchmarks/dnn/tensor/tensor_suite_test.go
+++ b/benchmarks/dnn/tensor/tensor_suite_test.go
@@ -3,7 +3,7 @@ package tensor
 import (
 	"testing"
 
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 )
 
diff --git a/benchmarks/dnn/training/crossentropy_test.go b/benchmarks/dnn/training/crossentropy_test.go
index c7b95aee..f1ac855b 100644
--- a/benchmarks/dnn/training/crossentropy_test.go
+++ b/benchmarks/dnn/training/crossentropy_test.go
@@ -1,7 +1,7 @@
 package training_test
 
 import (
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 	"github.com/sarchlab/mgpusim/v3/benchmarks/dnn/tensor"
 	"github.com/sarchlab/mgpusim/v3/benchmarks/dnn/training"
diff --git a/benchmarks/dnn/training/optimization/optimization_suite_test.go b/benchmarks/dnn/training/optimization/optimization_suite_test.go
index f82c77fb..3d18e7a9 100644
--- a/benchmarks/dnn/training/optimization/optimization_suite_test.go
+++ b/benchmarks/dnn/training/optimization/optimization_suite_test.go
@@ -3,7 +3,7 @@ package optimization
 import (
 	"testing"
 
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 )
 
diff --git a/benchmarks/dnn/training/optimization/sgd_test.go b/benchmarks/dnn/training/optimization/sgd_test.go
index 9210c185..791f45cc 100644
--- a/benchmarks/dnn/training/optimization/sgd_test.go
+++ b/benchmarks/dnn/training/optimization/sgd_test.go
@@ -2,7 +2,7 @@ package optimization
 
 import (
 	"github.com/golang/mock/gomock"
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 )
 
 var _ = Describe("SGD", func() {
diff --git a/benchmarks/dnn/training/softmaxcrossentropy_test.go b/benchmarks/dnn/training/softmaxcrossentropy_test.go
index bdb45eee..4ada055d 100644
--- a/benchmarks/dnn/training/softmaxcrossentropy_test.go
+++ b/benchmarks/dnn/training/softmaxcrossentropy_test.go
@@ -1,7 +1,7 @@
 package training
 
 import (
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 
 	"github.com/sarchlab/mgpusim/v3/benchmarks/dnn/tensor"
diff --git a/benchmarks/dnn/training/trainer_test.go b/benchmarks/dnn/training/trainer_test.go
index 87822975..127f96e5 100644
--- a/benchmarks/dnn/training/trainer_test.go
+++ b/benchmarks/dnn/training/trainer_test.go
@@ -2,7 +2,7 @@ package training
 
 import (
 	"github.com/golang/mock/gomock"
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	"github.com/sarchlab/mgpusim/v3/benchmarks/dnn/layers"
 	"github.com/sarchlab/mgpusim/v3/benchmarks/dnn/tensor"
 )
diff --git a/benchmarks/dnn/training/training_suite_test.go b/benchmarks/dnn/training/training_suite_test.go
index 6813eb6b..c3f9adf0 100644
--- a/benchmarks/dnn/training/training_suite_test.go
+++ b/benchmarks/dnn/training/training_suite_test.go
@@ -3,7 +3,7 @@ package training
 import (
 	"testing"
 
-	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 )