diff --git a/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/model/StakingModel.kt b/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/model/StakingModel.kt
index f4d1fdfd3e..a70dfc89f5 100644
--- a/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/model/StakingModel.kt
+++ b/features/staking/impl/src/main/java/com/tangem/features/staking/impl/presentation/model/StakingModel.kt
@@ -203,17 +203,19 @@ internal class StakingModel @Inject constructor(
     private val balancesToShow: List<StakingBalanceEntry>
         get() {
             val stakingBalance = cryptoCurrencyStatus.value.stakingBalance
-            return when (stakingBalance) {
-                is StakingBalance.Data.StakeKit -> {
-                    val invalidatedItems = invalidatePendingTransactionsUseCase(
-                        balanceItems = stakingBalance.balance.items,
+            return when (integration) {
+                is StakeKitIntegration -> {
+                    val balanceItems = (stakingBalance as? StakingBalance.Data.StakeKit)
+                        ?.balance?.items.orEmpty()
+                    invalidatePendingTransactionsUseCase(
+                        balanceItems = balanceItems,
                         stakingActions = stakingActions,
                         token = integration.token,
-                    ).getOrElse { emptyList() }
-                    invalidatedItems.toStakingBalanceEntries()
+                    ).getOrElse { emptyList() }.toStakingBalanceEntries()
+                }
+                is P2PEthPoolIntegration -> {
+                    (stakingBalance as? StakingBalance.Data.P2PEthPool)?.entries.orEmpty()
                 }
-                is StakingBalance.Data.P2PEthPool -> stakingBalance.entries
-                else -> emptyList()
             }
         }