Fix wallet connect issue

src/components/Modals/WalletConnect/index.tsx

@@ -18,23 +18,35 @@ import Image from 'next/image';
 import { useCoretimeApi, useRelayApi } from '@/contexts/apis';
 
 import styles from './index.module.scss';
+import { useEffect, useState } from 'react';
 
 interface WalletModalProps {
   open: boolean;
   onClose: () => void;
 }
 
 export const WalletModal = (props: WalletModalProps) => {
-  const { connect: connectContract, activeChain } = useInkathon();
+  const { connect: connectContract, activeChain, isConnected } = useInkathon();
   const { connectRelay } = useRelayApi();
   const { connectCoretime } = useCoretimeApi();
+
+  const [wallet, setWallet] = useState<SubstrateWallet | null>(null);
+
   const onConnect = async (wallet: SubstrateWallet) => {
+    setWallet(wallet);
     if (!connectContract) return;
     connectRelay();
     connectCoretime();
     connectContract(activeChain, wallet);
     props.onClose();
   };
+
+  useEffect(() => {
+    if (wallet) {
+      onConnect(wallet);
+    }
+  }, [isConnected]);
+
   return (
     <Dialog {...props} fullWidth maxWidth='sm'>
       <DialogTitle>Choose your wallet extension</DialogTitle>