diff --git a/invenio_theme_tugraz/deposits.py b/invenio_theme_tugraz/deposits.py
index bb25219..188b871 100644
--- a/invenio_theme_tugraz/deposits.py
+++ b/invenio_theme_tugraz/deposits.py
@@ -6,13 +6,11 @@
 # modify it under the terms of the MIT License; see LICENSE file for more
 # details.
 
+"""Routes for record-related pages provided by Invenio-App-RDM."""
 
 from flask import render_template
 from flask_login import login_required
-from invenio_app_rdm.records_ui.views.decorators import (
-    pass_draft,
-    pass_draft_files,
-)
+from invenio_app_rdm.records_ui.views.decorators import pass_draft, pass_draft_files
 from invenio_app_rdm.records_ui.views.deposits import (
     get_form_config,
     get_search_url,
diff --git a/invenio_theme_tugraz/ext.py b/invenio_theme_tugraz/ext.py
index fc07d54..1d9b53c 100644
--- a/invenio_theme_tugraz/ext.py
+++ b/invenio_theme_tugraz/ext.py
@@ -8,9 +8,10 @@
 
 """invenio module for TUGRAZ theme."""
 
+from invenio_theme_tugraz.deposits import deposit_create, deposit_edit
+
 from . import config
 from .views import index
-from invenio_theme_tugraz.deposits import deposit_create, deposit_edit
 
 
 class InvenioThemeTugraz(object):
diff --git a/setup.py b/setup.py
index db97bbe..7f7d6e2 100644
--- a/setup.py
+++ b/setup.py
@@ -18,6 +18,7 @@ history = open("CHANGES.rst").read()
 tests_require = [
     "pytest-invenio>=1.4.0",
     "invenio-app>=1.3.0,<2.0.0",
+    "invenio-app-rdm>=7.0.1,<8.0.0",
 ]
 
 # Should follow invenio-app-rdm