diff --git a/pages/404.tsx b/pages/404.tsx
index c327297..50f974b 100644
--- a/pages/404.tsx
+++ b/pages/404.tsx
@@ -1,7 +1,7 @@
import { NextPage } from 'next'
import { ErrorText } from '@utils/Constants'
-const NotFound: NextPage = () => {
+const NotFound: NextPage<{ message?: string }> = ({ message }) => {
return (
{
4
- {ErrorText[404]}
+ {message || ErrorText[404]}
diff --git a/pages/categories/[category].tsx b/pages/categories/[category].tsx
index 3277b8c..4f1cec3 100644
--- a/pages/categories/[category].tsx
+++ b/pages/categories/[category].tsx
@@ -26,7 +26,7 @@ const Category: NextPage = ({ data, query }) => {
useEffect(() => {
setNSFW(localStorage.nsfw)
}, [])
- if(!data || data.data.length === 0 || data.totalPage < Number(query.page)) return
+ if(!data || data.data.length === 0 || data.totalPage < Number(query.page)) return
return <>