diff --git a/tests/integration/admin/testinfra/config.go b/tests/integration/admin/testinfra/config.go
index b1f712f..e7f222d 100644
--- a/tests/integration/admin/testinfra/config.go
+++ b/tests/integration/admin/testinfra/config.go
@@ -5,6 +5,8 @@ package testinfra
 import (
 	"os"
 
+	_ "github.com/gogf/gf/contrib/drivers/pgsql/v2"
+
 	"github.com/gogf/gf/v2/frame/g"
 )
 
diff --git a/tests/integration/tenant/testinfra/config.go b/tests/integration/tenant/testinfra/config.go
index 146e18e..b3ed3d0 100644
--- a/tests/integration/tenant/testinfra/config.go
+++ b/tests/integration/tenant/testinfra/config.go
@@ -5,6 +5,8 @@ package testinfra
 import (
 	"os"
 
+	_ "github.com/gogf/gf/contrib/drivers/pgsql/v2"
+
 	"github.com/gogf/gf/v2/frame/g"
 )