diff --git a/server/src/main/java/org/opensearch/search/internal/ExitableDirectoryReader.java b/server/src/main/java/org/opensearch/search/internal/ExitableDirectoryReader.java
index 9187c4c07cba0..744649766be15 100644
--- a/server/src/main/java/org/opensearch/search/internal/ExitableDirectoryReader.java
+++ b/server/src/main/java/org/opensearch/search/internal/ExitableDirectoryReader.java
@@ -255,7 +255,7 @@ public int nextDoc() throws IOException {
 
         @Override
         public int advance(int target) throws IOException {
-            queryCancellation.checkCancelled();
+            checkAndThrowWithSampling();
             return super.advance(target);
         }
     }