diff --git a/tests/tests_test.go b/tests/tests_test.go
index 9a2970bd..4e667eae 100644
--- a/tests/tests_test.go
+++ b/tests/tests_test.go
@@ -5,6 +5,7 @@ import (
 	"math/rand"
 	"os"
 	"path/filepath"
+	"testing"
 	"time"
 
 	"gorm.io/driver/mysql"