fix integration tests on balance component with new watchAsset

1 files changed, 3 insertions, 3 deletions

diff --git a/ui/app/components/balance-component.js b/ui/app/components/balance-component.js
index 9af27f4ec..f85d1cdcd 100644
--- a/ui/app/components/balance-component.js
+++ b/ui/app/components/balance-component.js
@@ -34,8 +34,8 @@ function BalanceComponent () {
 BalanceComponent.prototype.render = function () {
   const props = this.props
   const { token, network, assetImages } = props
-  let imageUrl
-  if (token) imageUrl = assetImages[token.address]
+  const address = token && token.address
+  const imageUrl = assetImages && address ? assetImages[token.address] : undefined
 
   return h('div.balance-container', {}, [
 
@@ -46,7 +46,7 @@ BalanceComponent.prototype.render = function () {
     // }),
     h(Identicon, {
       diameter: 50,
-      address: token && token.address,
+      address,
       network,
       imageUrl,
     }),