fix(tokens): ensure valid token selection when updating a single token as well as all of them

Signed-off-by: Brandon McAnsh <git@bmcreations.dev>

Author: bmc08gt

services/opencode/src/main/kotlin/com/getcode/opencode/controllers/TokenController.kt

@@ -483,6 +483,7 @@ class TokenController @Inject constructor(
 
             if (tokenWithBalance != null) {
                 applyTokenUpdates(listOf(tokenWithBalance))
+                ensureValidTokenSelection()
             }
         } catch (e: Exception) {
             trace(
@@ -578,10 +579,10 @@ class TokenController @Inject constructor(
                 type = TraceType.Process
             )
             _state.update { it.copy(balances = it.balances + (token.address to newBalance)) }
-        }
 
-        scope.launch(Dispatchers.IO) {
-            updateTokenAccount(token.address)
+            scope.launch(Dispatchers.IO) {
+                updateTokenAccount(token.address)
+            }
         }
     }