diff --git a/src/Unit/Exchange/generateSwapCalldata.ts b/src/Unit/Exchange/generateSwapCalldata.ts
index a55d4a5..948d094 100644
--- a/src/Unit/Exchange/generateSwapCalldata.ts
+++ b/src/Unit/Exchange/generateSwapCalldata.ts
@@ -62,7 +62,7 @@ export async function generateSwapCalldataWithUnit({
 
   let path = SafeArray.from(arrayLikePath);
   const walletBalance = await getWalletBalance(
-    path.first().assetIn,
+    safeGet(assetToAddress, path.first().assetIn),
     receiverAddress,
     unit.provider
   );