diff --git a/domain/model/diamond_m/diamond.go b/domain/model/diamond_m/diamond.go
index 8a28e1ecc04f38a6cfa3b706cf315e0589e93844..d58d1ea73de142e8c61bfbcc0d5a14305eef8384 100644
--- a/domain/model/diamond_m/diamond.go
+++ b/domain/model/diamond_m/diamond.go
@@ -214,7 +214,7 @@ func (diamondAccountDetail *DiamondAccountDetail) PersistentNoInTransactional()
 	}
 
 	var newDiamondAccount DiamondAccount
-	if err := diamondAccountDetail.DB().Table(DiamondAccountDetail{}.TableName()).First(&newDiamondAccount, diamondAccountDetail.diamondAccount.ID).Error; err != nil {
+	if err := diamondAccountDetail.Db.First(&newDiamondAccount, diamondAccountDetail.diamondAccount.ID).Error; err != nil {
 		return myerr.WrapErr(err)
 	}
 
@@ -265,7 +265,7 @@ func (diamondAccountDetail *DiamondAccountDetail) Persistent() error {
 	}
 
 	var newDiamondAccount DiamondAccount
-	if err := diamondAccountDetail.DB().Table(DiamondAccountDetail{}.TableName()).First(&newDiamondAccount, diamondAccountDetail.diamondAccount.ID).Error; err != nil {
+	if err := diamondAccountDetail.Db.First(&newDiamondAccount, diamondAccountDetail.diamondAccount.ID).Error; err != nil {
 		return myerr.WrapErr(err)
 	}