From 236deba62c49d38c6b71c2e499d41f1c89fcb425 Mon Sep 17 00:00:00 2001
From: peachbits <matthew@edge.app>
Date: Wed, 27 Sep 2023 14:16:40 -0700
Subject: [PATCH] If swap requires a wrap function then correctly use the
 wrapped address

---
 src/swap/defi/defiUtils.ts | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/swap/defi/defiUtils.ts b/src/swap/defi/defiUtils.ts
index 6c8a2f3c..d520c05c 100644
--- a/src/swap/defi/defiUtils.ts
+++ b/src/swap/defi/defiUtils.ts
@@ -41,11 +41,14 @@ export const getInOutTokenAddresses = (
     ? undefined
     : allTokens[toTokenId]
 
-  const fromTokenAddress =
-    asMaybe(asContractLocation)(fromToken?.networkLocation)?.contractAddress ??
-    ''
-  const toTokenAddress =
-    asMaybe(asContractLocation)(toToken?.networkLocation)?.contractAddress ?? ''
+  const fromTokenAddress = isFromNativeCurrency
+    ? wrappedMainnetAddress
+    : asMaybe(asContractLocation)(fromToken?.networkLocation)
+        ?.contractAddress ?? ''
+  const toTokenAddress = isToNativeCurrency
+    ? wrappedMainnetAddress
+    : asMaybe(asContractLocation)(toToken?.networkLocation)?.contractAddress ??
+      ''
 
   const isFromWrappedCurrency =
     fromTokenAddress.toLowerCase() === wrappedMainnetAddress.toLowerCase()