fix(@wallet): delete selected account

status-im · Apr 25, 2023 · bdaf122 · bdaf122
1 parent c94997d
commit bdaf122
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/app/modules/main/wallet_section/overview/module.nim b/src/app/modules/main/wallet_section/overview/module.nim
@@ -95,9 +95,11 @@ proc setBalance(self: Module, tokens: seq[WalletTokenDto]) =
 
 # TODO(alaibe): replace with filter logic
 method switchAccount*(self: Module, accountIndex: int) =
+  var walletAccount = self.controller.getWalletAccount(accountIndex)
   self.currentAccountIndex = accountIndex
-
-  let walletAccount = self.controller.getWalletAccount(accountIndex)
+  if walletAccount.isNil:
+    self.currentAccountIndex = 0
+    walletAccount = self.controller.getWalletAccount(self.currentAccountIndex)
 
   let item = initItem(
     walletAccount.name,
@@ -133,4 +135,5 @@ proc onAccountAdded(self: Module, account: WalletAccountDto) =
   self.switchAccount(self.currentAccountIndex)
 
 proc onAccountRemoved(self: Module, account: WalletAccountDto) =
-  self.switchAccount(self.currentAccountIndex)
+  self.switchAccount(self.currentAccountIndex)
+