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