From 9cc1e8a6d867b7f0663c55b017b471132f6a719e Mon Sep 17 00:00:00 2001
From: Dan Finlay <dan@danfinlay.com>
Date: Wed, 18 Oct 2017 14:22:04 -0700
Subject: [PATCH] Refresh computed balances controller when restoring vault

---
 app/scripts/controllers/computed-balances.js | 4 ++++
 app/scripts/metamask-controller.js           | 3 ++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/app/scripts/controllers/computed-balances.js b/app/scripts/controllers/computed-balances.js
index 2479e1b3a..3479eae2b 100644
--- a/app/scripts/controllers/computed-balances.js
+++ b/app/scripts/controllers/computed-balances.js
@@ -25,6 +25,10 @@ class ComputedbalancesController {
     }
   }
 
+  forgetAllBalances () {
+    this.balances = {}
+  }
+
   _initBalanceUpdating () {
     const store = this.accountTracker.store.getState()
     this.addAnyAccountsFromStore(store)
diff --git a/app/scripts/metamask-controller.js b/app/scripts/metamask-controller.js
index b6a3749e4..b312106dd 100644
--- a/app/scripts/metamask-controller.js
+++ b/app/scripts/metamask-controller.js
@@ -473,7 +473,7 @@ module.exports = class MetamaskController extends EventEmitter {
   }
 
   forgetOldAccounts () {
-    const { accountTracker } = this
+    const { accountTracker, balancesController } = this
     let oldAccounts = []
     try {
       oldAccounts = Object.keys(accountTracker.store.getState().accounts)
@@ -481,6 +481,7 @@ module.exports = class MetamaskController extends EventEmitter {
       log.warn('Could not load old accounts to forget', e)
     }
     oldAccounts.forEach(addr => accountTracker.removeAccount(addr))
+    balancesController.forgetAllBalances()
   }
 
   selectFirstIdentity (vault) {