diff --git a/package.json b/package.json
index 7841dbc..b5bf7de 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@airdao/ui-library",
-  "version": "2.0.2",
+  "version": "2.0.3",
   "types": "./dist/types.d.ts",
   "exports": {
     ".": {
diff --git a/src/components/ConnectWallet/index.tsx b/src/components/ConnectWallet/index.tsx
index 6486dac..947fe26 100644
--- a/src/components/ConnectWallet/index.tsx
+++ b/src/components/ConnectWallet/index.tsx
@@ -38,6 +38,7 @@ const Option = ({
 
   const handleConnect = async (connector: Connector) => {
     try {
+      //TODO: fix this
       await connector.connect();
     } catch (error) {
       console.error('Failed to connect:', error);
diff --git a/src/components/Header/components/HeaderBody.tsx b/src/components/Header/components/HeaderBody.tsx
index d0626a5..ca21e9b 100644
--- a/src/components/Header/components/HeaderBody.tsx
+++ b/src/components/Header/components/HeaderBody.tsx
@@ -13,6 +13,7 @@ import cross from '../assets/cross.svg';
 import hamburgerIcon from '../assets/hamburger.svg';
 import { AddressInfo } from '../../AddressInfo';
 import { ConnectWalletModal } from '../../ConnectWalletModal';
+import { CONNECTOR_ICONS } from '../../../utils';
 
 export function HeaderBody({
   disconnect,
@@ -157,7 +158,11 @@ export function HeaderBody({
                 onClick={handleAddressInfo}
               >
                 <img
-                  src={currentConnector.icon}
+                  src={
+                    currentConnector.icon
+                      ? CONNECTOR_ICONS[currentConnector.icon]
+                      : ''
+                  }
                   alt={currentConnector.name + 'icon'}
                   className={s['header__address-icon']}
                 />