diff --git a/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/judgments/clickmodel/coec/CoecClickModel.java b/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/judgments/clickmodel/coec/CoecClickModel.java
index b50ceb0..e660472 100644
--- a/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/judgments/clickmodel/coec/CoecClickModel.java
+++ b/opensearch-search-quality-evaluation-plugin/src/main/java/org/opensearch/eval/judgments/clickmodel/coec/CoecClickModel.java
@@ -356,8 +356,7 @@ public Map<Integer, Double> getRankAggregatedClickThrough() throws Exception {
 
         }
 
-        // TODO: Should this be LTE or just LT? Is rank zero-based?
-        for(int rank = 0; rank <= parameters.getMaxRank(); rank++) {
+        for(int rank = 0; rank < parameters.getMaxRank(); rank++) {
 
             if(impressionCounts.containsKey(rank)) {