From e2a6af1ccb508491d9c526e2e92d026ba75fc79e Mon Sep 17 00:00:00 2001
From: Tangem <git@tangem.com>
Date: Thu, 26 Jun 2025 18:11:51 +0400
Subject: [PATCH] Updated on 2026-08-14

---
 .../staking/multi/DefaultMultiYieldBalanceFetcher.kt     | 2 +-
 .../presentation/state/helpers/StakingBalanceUpdater.kt  | 9 +++++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/data/staking/src/main/java/com/tangem/data/staking/multi/DefaultMultiYieldBalanceFetcher.kt b/data/staking/src/main/java/com/tangem/data/staking/multi/DefaultMultiYieldBalanceFetcher.kt
index 8019b9fcc8..8c21c90072 100644
--- a/data/staking/src/main/java/com/tangem/data/staking/multi/DefaultMultiYieldBalanceFetcher.kt
+++ b/data/staking/src/main/java/com/tangem/data/staking/multi/DefaultMultiYieldBalanceFetcher.kt
@@ -165,7 +165,7 @@ internal class DefaultMultiYieldBalanceFetcher @Inject constructor(
                 val yieldBalances = coroutineScope {
                     requests
                         // TODO: in the future, consider optimizing this part
-                        .chunked(size = 16) // StakeKitApi limitation: no more than 16 requests at the same time
+                        .chunked(size = 15) // StakeKitApi limitation: no more than 15 requests at the same time
                         .map {
                             async(dispatchers.io) { stakeKitApi.getMultipleYieldBalances(it).bind() }
                         }
diff --git a/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/state/helpers/StakingBalanceUpdater.kt b/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/state/helpers/StakingBalanceUpdater.kt
index 0bb0d545b7..e6c32aa82c 100644
--- a/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/state/helpers/StakingBalanceUpdater.kt
+++ b/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/state/helpers/StakingBalanceUpdater.kt
@@ -1,6 +1,7 @@
 package com.tangem.features.staking.impl.presentation.state.helpers
 
 import com.tangem.domain.staking.FetchActionsUseCase
+import com.tangem.domain.staking.FetchStakingYieldBalanceUseCase
 import com.tangem.domain.staking.model.stakekit.Yield
 import com.tangem.domain.staking.model.stakekit.action.StakingActionStatus
 import com.tangem.domain.tokens.FetchCurrencyStatusUseCase
@@ -22,6 +23,7 @@ internal class StakingBalanceUpdater @AssistedInject constructor(
     private val fetchActionsUseCase: FetchActionsUseCase,
     private val txHistoryContentUpdateEmitter: TxHistoryContentUpdateEmitter,
     private val fetchCurrencyStatusUseCase: FetchCurrencyStatusUseCase,
+    private val fetchStakingYieldBalanceUseCase: FetchStakingYieldBalanceUseCase,
     @DelayedWork private val coroutineScope: CoroutineScope,
     @Assisted private val userWallet: UserWallet,
     @Assisted private val cryptoCurrencyStatus: CryptoCurrencyStatus,
@@ -37,6 +39,13 @@ internal class StakingBalanceUpdater @AssistedInject constructor(
                         network = cryptoCurrencyStatus.currency.network,
                     )
                 },
+                async {
+                    fetchStakingYieldBalanceUseCase(
+                        userWalletId = userWallet.walletId,
+                        cryptoCurrency = cryptoCurrencyStatus.currency,
+                        isRefactoringEnabled = true,
+                    )
+                },
                 // we should update tx history and network for new balances
                 async {
                     fetchCurrencyStatus(delayMillis = BALANCE_UPDATE_DELAY)