diff --git a/oarepo_model_builder_drafts/invenio/__init__.py b/oarepo_model_builder_drafts/invenio/__init__.py
index cf48551..b7d64c6 100644
--- a/oarepo_model_builder_drafts/invenio/__init__.py
+++ b/oarepo_model_builder_drafts/invenio/__init__.py
@@ -18,4 +18,5 @@
     "drafts-record-communities-service-config": "templates/invenio_drafts_record_communities_service_config.py.jinja2",
     "drafts-record-search-options": "templates/invenio_drafts_record_search_options.py.jinja2",
     "drafts-record-facets": "templates/invenio_drafts_record_facets.py.jinja2",
+    "drafts-config": "templates/invenio_drafts_config.py.jinja2",
 }
diff --git a/oarepo_model_builder_drafts/invenio/invenio_drafts_config.py b/oarepo_model_builder_drafts/invenio/invenio_drafts_config.py
new file mode 100644
index 0000000..9fb5ace
--- /dev/null
+++ b/oarepo_model_builder_drafts/invenio/invenio_drafts_config.py
@@ -0,0 +1,7 @@
+from oarepo_model_builder.invenio.invenio_base import InvenioBaseClassPythonBuilder
+
+
+class InvenioDraftsConfigBuilder(InvenioBaseClassPythonBuilder):
+    TYPE = "invenio_drafts_config"
+    section = "config"
+    template = "drafts-config"
diff --git a/oarepo_model_builder_drafts/invenio/templates/invenio_drafts_config.py.jinja2 b/oarepo_model_builder_drafts/invenio/templates/invenio_drafts_config.py.jinja2
new file mode 100644
index 0000000..a1ed43b
--- /dev/null
+++ b/oarepo_model_builder_drafts/invenio/templates/invenio_drafts_config.py.jinja2
@@ -0,0 +1,7 @@
+{{ vars.record.class|imports }}
+
+OAREPO_PRIMARY_RECORD_SERVICE = {
+    {{ vars.record.class|base_name }}: "{{ vars.service_config.service_id }}"
+}
+
+{{ vars.config|extra_code }}
\ No newline at end of file
diff --git a/setup.cfg b/setup.cfg
index d14dbdd..b26fe3a 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = oarepo-model-builder-drafts
-version = 4.0.31
+version = 4.0.32
 description =
 authors = Ronald Krist <krist@cesnet.cz>
 readme = README.md
@@ -55,6 +55,9 @@ oarepo_model_builder.builders.draft:
     6000-invenio_conftest = oarepo_model_builder_drafts.tests.invenio_drafts_conftest:InvenioDraftsConftestBuilder
     6100-invenio_drafts_test_resource = oarepo_model_builder_drafts.tests.invenio_drafts_test_resources:InvenioDraftsTestResourcesBuilder
     6200-invenio_drafts_test_service = oarepo_model_builder_drafts.tests.invenio_drafts_test_services:InvenioDraftsTestServicesBuilder
+
+    7000-invenio_drafts_config = oarepo_model_builder_drafts.invenio.invenio_drafts_config:InvenioDraftsConfigBuilder
+
 oarepo_model_builder.builders.record:
     0030-invenio_drafts_parent_marshmallow = oarepo_model_builder_drafts.invenio.invenio_drafts_parent_marshmallow:InvenioDraftsParentMarshmallowBuilder