From e4e5bfb4c92591e0d255e30a56f382b863bc00a4 Mon Sep 17 00:00:00 2001
From: Eric Bailey <git@esb.lol>
Date: Wed, 3 Apr 2024 18:10:28 -0500
Subject: [PATCH] Cleanup

---
 src/state/session/index.tsx | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/state/session/index.tsx b/src/state/session/index.tsx
index d0313d553f..9578acec84 100644
--- a/src/state/session/index.tsx
+++ b/src/state/session/index.tsx
@@ -192,11 +192,11 @@ export function Provider({children}: React.PropsWithChildren<{}>) {
   )
 
   const setCurrentAccount = React.useCallback(
-    (account: SessionAccount, expired = false) => {
+    (account: SessionAccount) => {
       setStateAndPersist(s => {
         return {
           ...s,
-          currentAccount: expired ? undefined : account,
+          currentAccount: account,
           accounts: [account, ...s.accounts.filter(a => a.did !== account.did)],
         }
       })
@@ -277,8 +277,9 @@ export function Provider({children}: React.PropsWithChildren<{}>) {
           ({expired, refreshedAccount}) => {
             if (expired) {
               clearCurrentAccount()
+            } else {
+              setCurrentAccount(refreshedAccount)
             }
-            setCurrentAccount(refreshedAccount, expired)
           },
           {networkErrorCallback: clearCurrentAccount},
         ),
@@ -325,8 +326,9 @@ export function Provider({children}: React.PropsWithChildren<{}>) {
           ({expired, refreshedAccount}) => {
             if (expired) {
               clearCurrentAccount()
+            } else {
+              setCurrentAccount(refreshedAccount)
             }
-            setCurrentAccount(refreshedAccount, expired)
           },
           {networkErrorCallback: clearCurrentAccount},
         ),
@@ -375,7 +377,7 @@ export function Provider({children}: React.PropsWithChildren<{}>) {
               clearCurrentAccount()
             } else {
               __globalAgent = agent
-              setCurrentAccount(refreshedAccount, expired)
+              setCurrentAccount(refreshedAccount)
             }
           },
           {networkErrorCallback: clearCurrentAccount},