diff --git a/src/main/java/org/opensearch/eval/samplers/AllQueriesQuerySampler.java b/src/main/java/org/opensearch/eval/samplers/AllQueriesQuerySampler.java
index 8d6e61a..54a78b5 100644
--- a/src/main/java/org/opensearch/eval/samplers/AllQueriesQuerySampler.java
+++ b/src/main/java/org/opensearch/eval/samplers/AllQueriesQuerySampler.java
@@ -51,7 +51,7 @@ public String sample() throws Exception {
             queries.merge(ubiQuery.getUserQuery(), 1L, Long::sum);
 
             // Will be useful for paging once implemented.
-            if(queries.size() > parameters.getQuerySetSize()) {
+            if(queries.size() >= parameters.getQuerySetSize()) {
                 break;
             }