diff --git a/src/main/java/org/elasticsearch/index/cache/filter/support/CacheKeyFilter.java b/src/main/java/org/elasticsearch/index/cache/filter/support/CacheKeyFilter.java
index 6786128b65f..4d499169558 100644
--- a/src/main/java/org/elasticsearch/index/cache/filter/support/CacheKeyFilter.java
+++ b/src/main/java/org/elasticsearch/index/cache/filter/support/CacheKeyFilter.java
@@ -19,9 +19,11 @@
 
 package org.elasticsearch.index.cache.filter.support;
 
+import org.apache.lucene.index.AtomicReaderContext;
 import org.apache.lucene.index.IndexReader;
 import org.apache.lucene.search.DocIdSet;
 import org.apache.lucene.search.Filter;
+import org.apache.lucene.util.Bits;
 import org.elasticsearch.common.Unicode;
 
 import java.io.IOException;
@@ -86,8 +88,8 @@ public interface CacheKeyFilter {
         }
 
         @Override
-        public DocIdSet getDocIdSet(IndexReader reader) throws IOException {
-            return filter.getDocIdSet(reader);
+        public DocIdSet getDocIdSet(AtomicReaderContext context, Bits acceptDocs) throws IOException {
+            return filter.getDocIdSet(context, acceptDocs);
         }
 
         @Override