diff --git a/internal/model/diamond_m/repo.go b/internal/model/diamond_m/repo.go
index d4d28401a67e63d32d1d8f075ab007937605af1e..31a9edf9988f6e1ae024a473c07cd486275a889f 100644
--- a/internal/model/diamond_m/repo.go
+++ b/internal/model/diamond_m/repo.go
@@ -45,7 +45,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 err
 	}