diff --git a/src/fmu/sumo/explorer/objects/_document_collection.py b/src/fmu/sumo/explorer/objects/_document_collection.py
index 6d92c99d..daca736b 100644
--- a/src/fmu/sumo/explorer/objects/_document_collection.py
+++ b/src/fmu/sumo/explorer/objects/_document_collection.py
@@ -245,10 +245,10 @@ async def _next_batch_async(self) -> List[Dict]:
 
         return len(hits["hits"])
 
-    def _postprocess_batch(self, hits):
+    def _postprocess_batch(self, hits, pit):
         return
 
-    async def _postprocess_batch_async(self, hits):
+    async def _postprocess_batch_async(self, hits, pit):
         return
 
     def _init_query(self, doc_type: str, query: Dict = None) -> Dict: