diff --git a/invenio_app_rdm/records_ui/views/decorators.py b/invenio_app_rdm/records_ui/views/decorators.py
index c45fcf6fa..cd448b1d9 100644
--- a/invenio_app_rdm/records_ui/views/decorators.py
+++ b/invenio_app_rdm/records_ui/views/decorators.py
@@ -79,7 +79,7 @@ def view(**kwargs):
                 kwargs["draft"] = draft
                 kwargs["files_locked"] = (
                     record_service.config.lock_edit_published_files(
-                        record_service, g.identity, record=draft._record
+                        record_service, g.identity, draft=draft, record=draft._record
                     )
                 )
                 return f(**kwargs)