Bugfix: Improve 404 consistency

rileyjshaw · Sep 1, 2023 · b842110 · b842110
1 parent 426c70f
commit b842110
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/components/layout.js b/src/components/layout.js
@@ -44,12 +44,14 @@ const Layout = ({children, location}) => {
 
 	let {pathname} = location;
 	if (pathname.endsWith('/')) pathname = pathname.slice(0, -1);
+	const is404 = !location.key;
 
-	const isBlogPost = pathname.match(/\/blog\/.*[^0-9/]/);
-	if (pathname.startsWith('/blog'))
+	const isBlogPost = !is404 && pathname.match(/\/blog\/.*[^0-9/]/);
+	if (!404 && pathname.startsWith('/blog'))
 		pathname = isBlogPost ? '/blog/post' : '/blog';
 
-	const showPageHeader = !(isBlogPost || pathname.startsWith('/curate/'));
+	const showPageHeader =
+		is404 || !(isBlogPost || pathname.startsWith('/curate/'));
 	return (
 		<div
 			className={cn(