Substituted lastRound with lastFinalizedRound in check if needed catc…

…h-up peer.
LimeChain · Jan 28, 2025 · a0f5ff1 · a0f5ff1
1 parent 6306ee6
commit a0f5ff1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/com/limechain/grandpa/state/GrandpaSetState.java b/src/main/java/com/limechain/grandpa/state/GrandpaSetState.java
@@ -286,10 +286,10 @@ public CatchUpReqMessage initiateCatchUpRequestMessage(NeighbourMessage neighbou
 
         // If peer has the same voter set id
         if (neighbourMessage.getSetId().equals(setId)) {
-            BigInteger latestRound = roundCache.getLatestRoundNumber(setId);
 
             // Check if needed to catch-up peer
-            if (neighbourMessage.getRound().compareTo(latestRound.add(CATCH_UP_THRESHOLD)) >= 0) {
+            if (neighbourMessage.getRound().compareTo(
+                    lastFinalizedRound.getRoundNumber().add(CATCH_UP_THRESHOLD)) >= 0) {
                 log.log(Level.FINE, "Neighbor message indicates that the round of Peer " + peerId + " is ahead.");
 
                 catchUpReqMessage = CatchUpReqMessage.builder()