diff --git a/server/src/main/java/org/opensearch/gateway/TransportNodesGatewayStartedShardHelper.java b/server/src/main/java/org/opensearch/gateway/TransportNodesGatewayStartedShardHelper.java
index 5168cac6f920b..d08291de2eac5 100644
--- a/server/src/main/java/org/opensearch/gateway/TransportNodesGatewayStartedShardHelper.java
+++ b/server/src/main/java/org/opensearch/gateway/TransportNodesGatewayStartedShardHelper.java
@@ -232,6 +232,14 @@ public String toString() {
         }
     }
 
+    /**
+     * This class extends the {@link GatewayShardStarted} which contains all necessary shard metadata like
+     * allocationId and replication checkpoint. It also has DiscoveryNode which is needed by
+     * {@link PrimaryShardAllocator} and {@link PrimaryShardBatchAllocator} to make allocation decision.
+     * This class removes the dependency of
+     * {@link TransportNodesListGatewayStartedShards.NodeGatewayStartedShards} to make allocation decisions by
+     * {@link PrimaryShardAllocator} or {@link PrimaryShardBatchAllocator}.
+     */
     public static class NodeGatewayShardStarted extends GatewayShardStarted {
 
         private final DiscoveryNode node;