Fix creator staking balance calculation

dappforce · Dec 8, 2023 · 0acf055 · 0acf055
1 parent 009624d
commit 0acf055
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/utils/balance.ts b/src/utils/balance.ts
@@ -14,14 +14,12 @@ export const calculateBalanceForStaking = (
     lockId?: string
 ) => {
   if (!data) return BN_ZERO
-  const { freeBalance, lockedBalance, locks } = data
+  const { freeBalance, totalBalance, lockedBalance, locks } = data
 
   const stakingLockedBalance = locks?.find(({ id }) => id === lockId)
 
   if (!stakingLockedBalance ||
-      lockedBalance === stakingLockedBalance?.amount) return freeBalance
+      lockedBalance === stakingLockedBalance?.amount) return totalBalance
 
-  return new BN(freeBalance)
-      .add(new BN(lockedBalance)
-          .sub(new BN(stakingLockedBalance.amount)))
+  return new BN(totalBalance).sub(new BN(stakingLockedBalance.amount))
 }