From fa430cac6f63d2411ae79e93c0bbb28dc6274b6a Mon Sep 17 00:00:00 2001
From: Shiv Natani <shiv.natani@navi.com>
Date: Fri, 20 Dec 2024 13:49:17 +0530
Subject: [PATCH] NTP-21164 | Shiv Natani | check balance delay bug fix
 (#14278)

---
 .../payment/nativepayment/viewmodel/MPSViewModel.kt | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/android/navi-payment/src/main/java/com/navi/payment/nativepayment/viewmodel/MPSViewModel.kt b/android/navi-payment/src/main/java/com/navi/payment/nativepayment/viewmodel/MPSViewModel.kt
index ad4eaa0b62..50d433f52a 100644
--- a/android/navi-payment/src/main/java/com/navi/payment/nativepayment/viewmodel/MPSViewModel.kt
+++ b/android/navi-payment/src/main/java/com/navi/payment/nativepayment/viewmodel/MPSViewModel.kt
@@ -573,12 +573,6 @@ constructor(
                     "upi_lite_enabled" to "false"
                 )
             )
-            accountList.forEach {
-                accountListCheckBalanceUseCase.updateCheckBalanceStateForLinkedAccount(
-                    linkedAccountEntity = it,
-                    isAccountEligible = it.eligibilityState.isAccountEligible
-                )
-            }
             accountList
                 .filter { it.accountId.isAccountIdOfTypeUpiLite() }
                 .let {
@@ -607,7 +601,12 @@ constructor(
                         }
                     finalList.addAll(updateUpiAccountsRanking(it))
                 }
-
+            finalList.forEach {
+                accountListCheckBalanceUseCase.updateCheckBalanceStateForLinkedAccount(
+                    linkedAccountEntity = it,
+                    isAccountEligible = it.eligibilityState.isAccountEligible
+                )
+            }
             _bankAccountsState.update { BankAccountsState.AccountList(finalList) }
             _selectedBankAccount.update { getPreferredSelectedAccount(finalList) }
             _accountEligibilityMap.update {