From bf322356812990f741369eeec9e51645382c340b Mon Sep 17 00:00:00 2001
From: Balrambhai Sharma <sharma.balrambhai@navi.com>
Date: Tue, 25 Mar 2025 14:55:35 +0530
Subject: [PATCH] NTP-44548 | Check balance back press fix (#15507)

---
 .../account/linked/ui/LinkedAccountsScreen.kt         | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/android/navi-pay/src/main/kotlin/com/navi/pay/onboarding/account/linked/ui/LinkedAccountsScreen.kt b/android/navi-pay/src/main/kotlin/com/navi/pay/onboarding/account/linked/ui/LinkedAccountsScreen.kt
index 786f448545..0174043263 100644
--- a/android/navi-pay/src/main/kotlin/com/navi/pay/onboarding/account/linked/ui/LinkedAccountsScreen.kt
+++ b/android/navi-pay/src/main/kotlin/com/navi/pay/onboarding/account/linked/ui/LinkedAccountsScreen.kt
@@ -217,7 +217,8 @@ fun LinkedAccountsScreen(
     }
 
     when (uiState) {
-        LinkedAccountsScreenUiState.Loading -> RenderLinkedAccountsShimmerScreen()
+        LinkedAccountsScreenUiState.Loading ->
+            RenderLinkedAccountsShimmerScreen(onNavigationIconClick = onBackClick)
         is LinkedAccountsScreenUiState.SelfTransferScreen ->
             RenderSelfTransferNoneAndSingleAccountScreen(
                 isMultipleCreditAccountPresent =
@@ -780,9 +781,13 @@ fun RenderLinkedAccountsScreenBottomSheet(
 }
 
 @Composable
-private fun RenderLinkedAccountsShimmerScreen() {
+private fun RenderLinkedAccountsShimmerScreen(onNavigationIconClick: () -> Unit) {
     Column(modifier = Modifier.fillMaxWidth()) {
-        NaviPayHeader(title = "", onNavigationIconClick = {}, modifier = Modifier)
+        NaviPayHeader(
+            title = "",
+            onNavigationIconClick = onNavigationIconClick,
+            modifier = Modifier,
+        )
 
         Spacer(modifier = Modifier.height(8.dp))