diff --git a/model.go b/model.go
index d16dbac..dcc01d1 100644
--- a/model.go
+++ b/model.go
@@ -118,8 +118,9 @@ func DbOf(name string) *gorm.DB {
 		panic(errNoModelFound(name))
 	}
 	log.Debugf("start get db from model(%s)", name)
-	var ctx, cancelFunc = context.WithTimeout(context.Background(), 3*time.Second)
-	var db, err = gorm.NewSessionWithContext(m.db, ctx)
+	var ctx, cancelFunc = context.WithTimeout(m.db.Statement.Context, 3*time.Second)
+	var db = gorm.NewSessionWithContext(m.db, ctx)
+	var err = db.Error
 	if err != nil {
 		cancelFunc()
 		log.Warningf("new gorm session failed, err:[%s]", err)