From cf62ae006879c30c98ea861e15fb6b1489f929e6 Mon Sep 17 00:00:00 2001
From: Jeromy Cannon <jeromy@swirldslabs.com>
Date: Wed, 8 Jan 2025 21:00:09 +0000
Subject: [PATCH] return node client in refresh node client

Signed-off-by: Jeromy Cannon <jeromy@swirldslabs.com>
---
 src/commands/node/tasks.ts  | 3 +--
 src/core/account_manager.ts | 1 +
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/commands/node/tasks.ts b/src/commands/node/tasks.ts
index a3ba5dd4b..951f4939e 100644
--- a/src/commands/node/tasks.ts
+++ b/src/commands/node/tasks.ts
@@ -1328,8 +1328,7 @@ export class NodeCommandTasks {
       self.logger.info(`nodeId: ${nodeId}, config.newAccountNumber: ${config.newAccountNumber}`);
 
       if (config.existingNodeAliases.length > 1) {
-        await self.accountManager.refreshNodeClient(config.namespace, config.nodeAlias);
-        config.nodeClient = await this.accountManager.loadNodeClient(config.namespace);
+        config.nodeClient = await self.accountManager.refreshNodeClient(config.namespace, config.nodeAlias);
       }
 
       try {
diff --git a/src/core/account_manager.ts b/src/core/account_manager.ts
index 1f96f1fee..b0435bd3b 100644
--- a/src/core/account_manager.ts
+++ b/src/core/account_manager.ts
@@ -194,6 +194,7 @@ export class AccountManager {
       treasuryAccountInfo.privateKey,
       skipNodeAlias,
     );
+    return this._nodeClient;
   }
 
   /**