fix: fetch balances when wallet address is valid (#1778)

OffchainLabs · Jul 22, 2024 · 6cb1eea · 6cb1eea
1 parent f07b0c5
commit 6cb1eea
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/packages/arb-token-bridge-ui/src/hooks/useBalance.ts b/packages/arb-token-bridge-ui/src/hooks/useBalance.ts
@@ -1,5 +1,5 @@
 import { useCallback, useMemo } from 'react'
-import { BigNumber } from 'ethers'
+import { BigNumber, utils } from 'ethers'
 import useSWR, {
   useSWRConfig,
   unstable_serialize,
@@ -40,10 +40,13 @@ const merge: Middleware = (useSWRNext: SWRHook) => {
 }
 
 const useBalance = ({ chainId, walletAddress }: UseBalanceProps) => {
-  const walletAddressLowercased = useMemo(
-    () => walletAddress?.toLowerCase(),
-    [walletAddress]
-  )
+  const walletAddressLowercased = useMemo(() => {
+    // use balances for the wallet address only if it's valid
+    if (!walletAddress || !utils.isAddress(walletAddress)) {
+      return undefined
+    }
+    return walletAddress.toLowerCase()
+  }, [walletAddress])
 
   const queryKey = useCallback(
     (type: 'eth' | 'erc20') => {