diff --git a/dtmgrpc/type.go b/dtmgrpc/type.go
index 42e7eb8..3990261 100644
--- a/dtmgrpc/type.go
+++ b/dtmgrpc/type.go
@@ -12,6 +12,7 @@ import (
 	"github.com/yedf/dtm/dtmcli"
 	"github.com/yedf/dtm/dtmcli/dtmimp"
 	"github.com/yedf/dtm/dtmgrpc/dtmgimp"
+	"github.com/yedf/dtmdriver"
 	emptypb "google.golang.org/protobuf/types/known/emptypb"
 )
 
@@ -32,3 +33,7 @@ func SetCurrentDBType(dbType string) {
 func GetCurrentDBType() string {
 	return dtmcli.GetCurrentDBType()
 }
+
+func UseDriver(driverName string) error {
+	return dtmdriver.Use(driverName)
+}
diff --git a/dtmgrpc/type_test.go b/dtmgrpc/type_test.go
index 2192d63..9440905 100644
--- a/dtmgrpc/type_test.go
+++ b/dtmgrpc/type_test.go
@@ -24,4 +24,7 @@ func TestType(t *testing.T) {
 	old := GetCurrentDBType()
 	SetCurrentDBType(dtmcli.DBTypeMysql)
 	SetCurrentDBType(old)
+
+	err = UseDriver("default")
+	assert.Nil(t, err)
 }