diff --git a/lucene/misc/src/java/org/apache/lucene/search/DiversifiedTopDocsCollector.java b/lucene/misc/src/java/org/apache/lucene/search/DiversifiedTopDocsCollector.java
index 6b756224439..1a087d47cf5 100644
--- a/lucene/misc/src/java/org/apache/lucene/search/DiversifiedTopDocsCollector.java
+++ b/lucene/misc/src/java/org/apache/lucene/search/DiversifiedTopDocsCollector.java
@@ -124,10 +124,7 @@ public abstract class DiversifiedTopDocsCollector extends
     // a leaf reader value when looking up keys
     int leafDocID = addition.doc - docBase;
     long value;
-    if (keys.docID() < leafDocID) {
-      keys.advance(leafDocID);
-    }
-    if (keys.docID() == leafDocID) {
+    if (keys.advanceExact(leafDocID)) {
       value = keys.longValue();
     } else {
       value = 0;