diff --git a/docs/source/conf.py b/docs/source/conf.py
index 4140dc4381..5624e36ed7 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -551,6 +551,7 @@ def setup(app):
 # https://github.com/kedro-org/kedro/issues/1772
 mermaid_output_format = "png"
 # https://github.com/mermaidjs/mermaid.cli#linux-sandbox-issue
+# https://github.com/mermaid-js/mermaid-cli/issues/544
 mermaid_params = ["-p", here / "puppeteer-config.json", "-s", "2"]
 # https://github.com/kedro-org/kedro/issues/2451
 mermaid_version = mermaid_init_js = ""
diff --git a/docs/source/puppeteer-config.json b/docs/source/puppeteer-config.json
index 3201af7b73..5adc866e94 100644
--- a/docs/source/puppeteer-config.json
+++ b/docs/source/puppeteer-config.json
@@ -1,3 +1,4 @@
 {
-  "args": ["--no-sandbox"]
+  "args": ["--no-sandbox"],
+  "headless": "old"
 }