From e3044ff82c6938ed7ccd7b8b06f71c27fb538b26 Mon Sep 17 00:00:00 2001
From: pedromcunha <pedro.spinsomewebs@gmail.com>
Date: Tue, 24 Sep 2024 15:32:38 -0400
Subject: [PATCH] Fix setting chain filter properly between token and chain ui

---
 .changeset/pink-countries-laugh.md                           | 5 +++++
 .../ui/src/components/common/TokenSelector/TokenSelector.tsx | 4 ++++
 2 files changed, 9 insertions(+)
 create mode 100644 .changeset/pink-countries-laugh.md

diff --git a/.changeset/pink-countries-laugh.md b/.changeset/pink-countries-laugh.md
new file mode 100644
index 00000000..ec2a807e
--- /dev/null
+++ b/.changeset/pink-countries-laugh.md
@@ -0,0 +1,5 @@
+---
+'@reservoir0x/relay-kit-ui': patch
+---
+
+Fix token selector balance display
diff --git a/packages/ui/src/components/common/TokenSelector/TokenSelector.tsx b/packages/ui/src/components/common/TokenSelector/TokenSelector.tsx
index 07ee0a6e..7f284f18 100644
--- a/packages/ui/src/components/common/TokenSelector/TokenSelector.tsx
+++ b/packages/ui/src/components/common/TokenSelector/TokenSelector.tsx
@@ -359,6 +359,10 @@ const TokenSelector: FC<TokenSelectorProps> = ({
     setTokenSelectorStep(TokenSelectorStep.SetCurrency)
     setTokenSearchInput('')
     setChainSearchInput('')
+    setChainFilter({
+      id: undefined,
+      name: 'All'
+    })
   }, [])
 
   const selectToken = useCallback(