diff --git a/layouts/404.html b/layouts/404.html
index 06ca617..f0b35fb 100644
--- a/layouts/404.html
+++ b/layouts/404.html
@@ -1,10 +1,19 @@
+<!DOCTYPE html>
+{{- partial "docs/shared" -}}
+<html>
 
-{{ define "main"}}
-<main id="main">
-  <div>
-   <h1 id="title"><a href="{{ "/" }}">Go Home</a></h1>
-  </div>
-</main>
-{{ end }}
+<head>
+  {{ partial "docs/html-head" . }}
+  {{ partial "docs/inject/head" . }}
+</head>
 
-{{ define "toc" }} {{ end }}
\ No newline at end of file
+<body>
+  <main class="flex justify-center">
+      <h1>404 Not Found</h1>
+  </main>
+
+  {{ partial "docs/inject/body" . }}
+  {{ template "_internal/google_analytics_async.html" . }}
+</body>
+
+</html>
diff --git a/layouts/docs/baseof.html b/layouts/docs/baseof.html
index 805e060..9623041 100644
--- a/layouts/docs/baseof.html
+++ b/layouts/docs/baseof.html
@@ -32,10 +32,10 @@
     </div>
 
     {{ template "toc" . }}
-
-    {{ partial "docs/inject/body" . }}
-    {{ template "_internal/google_analytics_async.html" . }}
   </main>
+
+  {{ partial "docs/inject/body" . }}
+  {{ template "_internal/google_analytics_async.html" . }}
 </body>
 
 </html>
