diff --git a/docs/main.py b/docs/main.py
index 3e4d82d..920ec41 100644
--- a/docs/main.py
+++ b/docs/main.py
@@ -1 +1 @@
-from nomad.mkdocs import define_env  # noqa: F401
+from nomad_docs import define_env  # noqa: F401
diff --git a/pyproject.toml b/pyproject.toml
index 1ca5faa..b848da0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -51,12 +51,13 @@ dev = [
     "pytest",
     "ruff",
     "structlog",
-    "mkdocs", 
+    "nomad-docs>=0.1.3",
+    "mkdocs",
     "mkdocs-material>=9.0",
     "pymdown-extensions",
     "mkdocs-click",
     "mkdocs-macros-plugin>=1.0",
-    "pydantic>=2.0,<2.11"
+    "pydantic>=2.0,<2.11",
 ]
 
 [tool.ruff]
@@ -106,6 +107,7 @@ select = [
 
 ignore = [
     "F403", # 'from module import *' used; unable to detect undefined names
+    "PLC0415", # `import` should be at the top-level of a file
 ]
 
 fixable = ["ALL"]