fix: store account address after login

Author: pcarranzav

packages/hypergraph/src/connect/login.ts

@@ -65,7 +65,7 @@ export async function signup(
     throw new Error('Error creating identity');
   }
   storeKeys(storage, accountAddress, keys);
-
+  storeAccountAddress(storage, accountAddress);
   return {
     accountAddress,
     keys,
@@ -94,6 +94,7 @@ export async function restoreKeys(
     const { ciphertext, nonce } = keyBox;
     const keys = await decryptIdentity(signer, ciphertext, nonce);
     storeKeys(storage, accountAddress, keys);
+    storeAccountAddress(storage, accountAddress);
     return {
       accountAddress,
       keys,
@@ -157,9 +158,6 @@ export async function login({
     console.log('receipt', receipt);
   }
   const accountAddress = smartAccountWalletClient.account.address;
-  if (accountAddressFromStorage === undefined) {
-    storeAccountAddress(storage, accountAddress);
-  }
   // const keys = loadKeys(storage, accountAddress);
   let authData: {
     accountAddress: Address;