From 21665144b331cb3d4db00fac8cacf4c6652d2d54 Mon Sep 17 00:00:00 2001
From: Christopher Howard <derhowiedesigns@gmail.com>
Date: Fri, 2 Aug 2024 12:32:48 -0400
Subject: [PATCH] import screen doesn't show account balance in USD shows $0
 (#1634)

Co-authored-by: Daniel Sinclair <4412473+DanielSinclair@users.noreply.github.com>
---
 src/entries/popup/hooks/useWalletsSummary.ts | 17 ++++++-----------
 1 file changed, 6 insertions(+), 11 deletions(-)

diff --git a/src/entries/popup/hooks/useWalletsSummary.ts b/src/entries/popup/hooks/useWalletsSummary.ts
index a9662967d7..05ec8b812c 100644
--- a/src/entries/popup/hooks/useWalletsSummary.ts
+++ b/src/entries/popup/hooks/useWalletsSummary.ts
@@ -1,19 +1,14 @@
 import { useMemo } from 'react';
 import { Address } from 'viem';
 
-import {
-  AVAX_AVALANCHE_ADDRESS,
-  BNB_MAINNET_ADDRESS,
-  ETH_ADDRESS,
-  MATIC_MAINNET_ADDRESS,
-  SupportedCurrencyKey,
-} from '~/core/references';
+import { SupportedCurrencyKey } from '~/core/references';
 import {
   AddySummary,
   useAddysSummary,
 } from '~/core/resources/addys/addysSummary';
 import { useCurrentCurrencyStore } from '~/core/state';
 import { ParsedAsset } from '~/core/types/assets';
+import { ChainId } from '~/core/types/chains';
 import {
   add,
   convertAmountAndPriceToNativeDisplay,
@@ -61,7 +56,7 @@ const parseAddressSummary = ({
   }).amount;
   const ethCurrencyBalance = convertAmountAndPriceToNativeDisplay(
     ethBalance || 0,
-    nativeAssets?.[`${ETH_ADDRESS}_1`]?.price?.value || 0,
+    nativeAssets?.[ChainId.mainnet]?.price?.value || 0,
     currentCurrency,
   ).amount;
 
@@ -70,7 +65,7 @@ const parseAddressSummary = ({
   }).amount;
   const bnbCurrencyBalance = convertAmountAndPriceToNativeDisplay(
     bnbBalance || 0,
-    nativeAssets?.[`${BNB_MAINNET_ADDRESS}_1`]?.price?.value || 0,
+    nativeAssets?.[ChainId.bsc]?.price?.value || 0,
     currentCurrency,
   ).amount;
 
@@ -82,7 +77,7 @@ const parseAddressSummary = ({
   ).amount;
   const maticCurrencyBalance = convertAmountAndPriceToNativeDisplay(
     maticBalance || 0,
-    nativeAssets?.[`${MATIC_MAINNET_ADDRESS}_1`]?.price?.value || 0,
+    nativeAssets?.[ChainId.polygon]?.price?.value || 0,
     currentCurrency,
   ).amount;
 
@@ -91,7 +86,7 @@ const parseAddressSummary = ({
   }).amount;
   const avaxCurrencyBalance = convertAmountAndPriceToNativeDisplay(
     avaxBalance || 0,
-    nativeAssets?.[`${AVAX_AVALANCHE_ADDRESS}_1`]?.price?.value || 0,
+    nativeAssets?.[ChainId.avalanche]?.price?.value || 0,
     currentCurrency,
   ).amount;