diff --git a/recordext/functions/set_record_documents.py b/recordext/functions/set_record_documents.py
index <HASH>..<HASH> 100644
--- a/recordext/functions/set_record_documents.py
+++ b/recordext/functions/set_record_documents.py
@@ -24,7 +24,7 @@ import os
 from invenio.base.utils import toposort_depends
 from invenio.modules.documents import api
 from invenio.modules.documents.tasks import set_document_contents
-from invenio.modules.pidstore.recordext.functions import reserve_recid
+from invenio_records.recordext.functions import reserve_recid
 from invenio_records.signals import before_record_insert
 from invenio_records.utils import name_generator
pidstore: removal of recid provider

* INCOMPATIBLE Removes recid provider, reserve recid function and
* datacite tasks as they were ported to invenio-records package.
  (addresses inveniosoftware/invenio-records#6)