From d0463b3aad65fead65bb18bcde298c704e2ff574 Mon Sep 17 00:00:00 2001
From: harry <53987565+h5law@users.noreply.github.com>
Date: Wed, 6 Dec 2023 01:49:26 +0000
Subject: [PATCH] chore: address final comments

---
 x/pocketcore/types/session.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/x/pocketcore/types/session.go b/x/pocketcore/types/session.go
index 8f387cb16..d2c4fde13 100644
--- a/x/pocketcore/types/session.go
+++ b/x/pocketcore/types/session.go
@@ -117,6 +117,8 @@ func NewSessionNodes(
 ) (sessionNodes SessionNodes, err sdk.Error) {
 	// retrieve the enforce max chains flag's value from the codec
 	isEnforceMaxChains := ModuleCdc.IsAfterEnforceMaxChainsUpgrade(ctx.BlockHeight())
+	// retrieve the max chains value from the sessionCtx
+	nodeMaxChains := keeper.MaxChains(sessionCtx)
 	// all nodesAddrs at session genesis
 	nodesAddrs, totalNodes := keeper.GetValidatorsByChain(sessionCtx, chain)
 	// validate nodesAddrs
@@ -151,7 +153,7 @@ func NewSessionNodes(
 		lenNodeChains := int64(len(node.GetChains()))
 		// if not found or jailed or is overstaked to chains
 		if node == nil ||
-			(isEnforceMaxChains && lenNodeChains > keeper.MaxChains(ctx)) ||
+			(isEnforceMaxChains && lenNodeChains > nodeMaxChains) ||
 			node.IsJailed() ||
 			!NodeHasChain(chain, node) ||
 			sessionNodes.Contains(node.GetAddress()) {