diff --git a/docs/assets/js/hljs.js b/docs/assets/js/hljs.js
index 6f9098ac1..56159c4c9 100644
--- a/docs/assets/js/hljs.js
+++ b/docs/assets/js/hljs.js
@@ -1,3 +1,3 @@
-document.addEventListener('DOMContentLoaded', (event) => {
+window.document$.subscribe(() => {
     hljs.highlightAll();
 });
diff --git a/mkdocs.yml b/mkdocs.yml
index b39c00e3f..bfd21dc99 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -28,7 +28,6 @@ theme:
         name: Switch to light mode
   features:
     - navigation.instant
-    - navigation.instant.prefetch
     - content.code.copy
     - navigation.footer
     - content.action.edit
@@ -55,7 +54,7 @@ extra:
       link: https://join.slack.com/t/codeigniterchat/shared_invite/zt-244xrrslc-l_I69AJSi5y2a2RVN~xIdQ
       name: Slack
 
-
+site_url: https://shield.codeigniter.com/
 repo_url: https://github.com/codeigniter4/shield
 edit_uri: edit/develop/docs/
 copyright: Copyright © 2023 CodeIgniter Foundation.