Update of the refresh method.

Trust/Tokens/Types/TokensDataStore.swift

@@ -109,7 +109,6 @@ class TokensDataStore {
                 switch result {
                 case .success(let response):
                     self.update(tokens: response)
-                    self.refreshEthBalance()
                     self.refreshBalance()
                 case .failure: break
                 }
@@ -139,22 +138,20 @@ class TokensDataStore {
                 }
                 count += 1
                 if count == updateTokens.count {
-                   self.updateDelegate()
+                    //We should use prommis kit.
+                    self.getBalanceCoordinator.getEthBalance(for: self.account.address) {  [weak self] result in
+                        guard let `self` = self else { return }
+                        switch result {
+                        case .success(let balance):
+                            self.update(token: self.objects.first (where: { $0.contract == self.ethToken.contract })!, action: .value(balance.value))
+                            self.updateDelegate()
+                        case .failure: break
+                        }
+                    }
                 }
             }
         }
     }
-    func refreshEthBalance() {
-        getBalanceCoordinator.getEthBalance(for: account.address) {  [weak self] result in
-            guard let `self` = self else { return }
-            switch result {
-            case .success(let balance):
-                self.update(token: self.objects.first (where: { $0.contract == self.ethToken.contract })!, action: .value(balance.value))
-                self.updateDelegate()
-            case .failure: break
-            }
-        }
-    }
     func updateDelegate() {
         tokensModel.value = enabledObject
         let tokensViewModel = TokensViewModel( tokens: enabledObject, tickers: tickers )

Trust/Transactions/Coordinators/BalanceCoordinator.swift

@@ -25,7 +25,7 @@ class BalanceCoordinator {
     init(account: Wallet, storage: TokensDataStore) {
         self.account = account
         self.storage = storage
-        self.storage.refreshEthBalance()
+        self.storage.refreshBalance()
         storage.tokensModel.subscribe {[weak self] tokensModel in
             guard let tokens = tokensModel, let eth = tokens.first(where: { $0.contract == "0x" }) else {
                 return
@@ -37,7 +37,7 @@ class BalanceCoordinator {
         }
     }
     func refresh() {
-        self.storage.refreshEthBalance()
+        self.storage.refreshBalance()
     }
     func update() {
         delegate?.didUpdate(viewModel: viewModel)