From b05a6f89cb2c4de1e53d96f8cac01653a8165555 Mon Sep 17 00:00:00 2001
From: frankiebee <frankie.diamond@gmail.com>
Date: Tue, 26 Sep 2017 18:19:44 -0700
Subject: [PATCH] fix tests

---
 test/unit/tx-controller-test.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/unit/tx-controller-test.js b/test/unit/tx-controller-test.js
index dd8b48684..9ffba31ee 100644
--- a/test/unit/tx-controller-test.js
+++ b/test/unit/tx-controller-test.js
@@ -25,7 +25,7 @@ describe('Transaction Controller', function () {
       networkStore: new ObservableStore(currentNetworkId),
       txHistoryLimit: 10,
       blockTracker: { getCurrentBlock: noop, on: noop, once: noop },
-      accountTracker: { getState: noop },
+      accountTracker: { store: {getState: noop} },
       signTransaction: (ethTx) => new Promise((resolve) => {
         ethTx.sign(privKey)
         resolve()
@@ -385,7 +385,7 @@ describe('Transaction Controller', function () {
 
   describe('#getBalance', function () {
     it('gets balance', function () {
-      sinon.stub(txController.ethStore, 'getState').callsFake(() => {
+      sinon.stub(txController.accountTracker.store, 'getState').callsFake(() => {
         return {
           accounts: {
             '0x1678a085c290ebd122dc42cba69373b5953b831d': {