From a04348e1426d2dc9e86a99c14a523a22a706e5b7 Mon Sep 17 00:00:00 2001
From: Florin Dzeladini <florin.dzeladini@consensys.net>
Date: Tue, 28 Jan 2025 14:43:13 +0100
Subject: [PATCH] chore: fix comments

---
 packages/wallet-ui/src/App.tsx                     | 9 ++++++++-
 packages/wallet-ui/src/services/useStarkNetSnap.ts | 2 +-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/packages/wallet-ui/src/App.tsx b/packages/wallet-ui/src/App.tsx
index d0cee58f..e59b1ff8 100644
--- a/packages/wallet-ui/src/App.tsx
+++ b/packages/wallet-ui/src/App.tsx
@@ -25,7 +25,8 @@ import { DeployModal } from 'components/ui/organism/DeployModal';
 library.add(fas, far);
 
 function App() {
-  const { initSnap, getWalletData, checkConnection } = useStarkNetSnap();
+  const { initSnap, getWalletData, checkConnection, loadLocale } =
+    useStarkNetSnap();
   const { connected, forceReconnect, provider } = useAppSelector(
     (state) => state.wallet,
   );
@@ -64,6 +65,12 @@ function App() {
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [networks.activeNetwork, provider]);
 
+  useEffect(() => {
+    if (connected) {
+      loadLocale();
+    }
+  }, [connected, loadLocale]);
+
   const loading = loader.isLoading;
   return (
     <ThemeProvider theme={theme}>
diff --git a/packages/wallet-ui/src/services/useStarkNetSnap.ts b/packages/wallet-ui/src/services/useStarkNetSnap.ts
index f3074728..e10c93ec 100644
--- a/packages/wallet-ui/src/services/useStarkNetSnap.ts
+++ b/packages/wallet-ui/src/services/useStarkNetSnap.ts
@@ -257,7 +257,6 @@ export const useStarkNetSnap = () => {
       dispatch(setActiveNetwork(idx));
       const chainId = net.chainId;
       await getWalletData(chainId, nets);
-      await loadLocale();
     } catch (err: any) {
       if (err.code && err.code === 4100) {
         const toastr = new Toastr();
@@ -1002,6 +1001,7 @@ export const useStarkNetSnap = () => {
 
   return {
     connectToSnap,
+    loadLocale,
     getNetworks,
     getAccounts,
     addAccount,