diff --git a/internal/data/repoimpl/order.go b/internal/data/repoimpl/order.go
index d594f92..0a42cdb 100644
--- a/internal/data/repoimpl/order.go
+++ b/internal/data/repoimpl/order.go
@@ -19,16 +19,16 @@ import (
 // OrderRepoImpl .
 type OrderRepoImpl struct {
 	Base[model.Order, bo.OrderBo]
-	db *data.GormDb
+	db *data.Db
 }
 
 // NewOrderRepoImpl .
-func NewOrderRepoImpl(db *data.GormDb) repo.OrderRepo {
+func NewOrderRepoImpl(db *data.Db) repo.OrderRepo {
 	return &OrderRepoImpl{db: db}
 }
 
 func (p *OrderRepoImpl) DB(ctx context.Context) *gorm.DB {
-	return p.db.Client.WithContext(ctx).Model(model.Order{})
+	return p.db.DB(ctx).Model(model.Order{})
 }
 
 func (p *OrderRepoImpl) FindInBatches(ctx context.Context, w *bo.FindInBatchesUseBo, fun func(ctx context.Context, rows []*bo.OrderBo) error) error {