diff --git a/pkg/machinepool/validations/disk_size_test.go b/pkg/machinepool/validations/disk_size_test.go
index 4f0a794..172f000 100644
--- a/pkg/machinepool/validations/disk_size_test.go
+++ b/pkg/machinepool/validations/disk_size_test.go
@@ -1,5 +1,10 @@
 package validations
 
+import (
+	. "github.com/onsi/ginkgo/v2/dsl/table"
+	. "github.com/onsi/gomega"
+)
+
 var _ = DescribeTable("getAWSVolumeMaxSize",
 	func(testSpec struct {
 		name string