diff --git a/oarepo_model_builder_drafts/datatypes/components/draft.py b/oarepo_model_builder_drafts/datatypes/components/draft.py
index 491c769..a3f0289 100644
--- a/oarepo_model_builder_drafts/datatypes/components/draft.py
+++ b/oarepo_model_builder_drafts/datatypes/components/draft.py
@@ -71,7 +71,7 @@ def process_links(self, datatype, section: Section, **kwargs):
                     name="self",
                     link_class="ConditionalLink",
                     link_args=[
-                        "cond=is_record",
+                        "cond=is_published_record",
                         f'if_=RecordLink("{{+api}}{url_prefix}{{id}}")',
                         f'else_=RecordLink("{{+api}}{url_prefix}{{id}}/draft")',
                     ],
@@ -79,7 +79,7 @@ def process_links(self, datatype, section: Section, **kwargs):
                         Import("invenio_records_resources.services.ConditionalLink"),
                         Import("invenio_records_resources.services.RecordLink"),
                         Import(
-                            "invenio_drafts_resources.services.records.config.is_record"
+                            "oarepo_runtime.records.is_published_record"
                         ),
                     ],
                 ),
@@ -87,7 +87,7 @@ def process_links(self, datatype, section: Section, **kwargs):
                     name="self_html",
                     link_class="ConditionalLink",
                     link_args=[
-                        "cond=is_record",
+                        "cond=is_published_record",
                         f'if_=RecordLink("{{+ui}}{html_url_prefix}{{id}}")',
                         f'else_=RecordLink("{{+ui}}{html_url_prefix}{{id}}/edit")',
                     ],
@@ -95,7 +95,7 @@ def process_links(self, datatype, section: Section, **kwargs):
                         Import("invenio_records_resources.services.ConditionalLink"),
                         Import("invenio_records_resources.services.RecordLink"),
                         Import(
-                            "invenio_drafts_resources.services.records.config.is_record"
+                            "oarepo_runtime.records.is_published_record"
                         ),
                     ],
                 ),
diff --git a/setup.cfg b/setup.cfg
index ce8fbcd..f975595 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = oarepo-model-builder-drafts
-version = 4.0.22
+version = 4.0.23
 description =
 authors = Ronald Krist <krist@cesnet.cz>
 readme = README.md