From 33cf25ac4c25f8df43ee113cf716f77e5da75883 Mon Sep 17 00:00:00 2001
From: Tangem <git@tangem.com>
Date: Tue, 10 Oct 2023 11:23:03 +0300
Subject: [PATCH] Updated on 2026-08-14

---
 .../presentation/wallet/viewmodels/WalletStateCache.kt       | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/viewmodels/WalletStateCache.kt b/features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/viewmodels/WalletStateCache.kt
index 489b6dd666..0c81b8e3ee 100644
--- a/features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/viewmodels/WalletStateCache.kt
+++ b/features/wallet/impl/src/main/java/com/tangem/feature/wallet/presentation/wallet/viewmodels/WalletStateCache.kt
@@ -20,9 +20,10 @@ internal object WalletStateCache {
         states[userWalletId] = state
     }
 
-    fun updateAll(func: (WalletState.ContentState.() -> WalletState.ContentState)) {
+    /** Update all content states  */
+    fun updateAll(block: (WalletState.ContentState.() -> WalletState.ContentState)) {
         states.keys.forEach {
-            states[it] = func(states[it]!!)
+            states[it] = block(requireNotNull(states[it]))
         }
     }
 }
\ No newline at end of file