1 files changed, 1 insertions, 12 deletions
diff --git a/api/main.py b/api/main.py
index 8be260d..1deeca7 100644
--- a/api/main.py
+++ b/api/main.py
@@ -1,13 +1,11 @@
 from fastapi import FastAPI, Depends, Request, Path
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse, RedirectResponse
+from fastapi.responses import RedirectResponse
 from api.routes.auth_routes import router as auth_router
 from api.routes.links_routes import router as links_router
 from api.routes.user_routes import router as user_router
 from api.routes.log_routes import router as log_router
 from typing import Annotated
-from fastapi.exceptions import HTTPException
-from starlette.status import HTTP_404_NOT_FOUND
 
 from api.util.db_dependency import get_db
 from api.util.log import log
@@ -65,12 +63,3 @@ async def log_redirect(
         log(link, ip, user_agent)
         db.close()
         return RedirectResponse(url=link_record.redirect_link)
-
-
-# Custom handler for 404 errors
-@app.exception_handler(HTTP_404_NOT_FOUND)
-async def custom_404_handler(request: Request, exc: HTTPException):
-    return JSONResponse(
-        status_code=404,
-        content={"message": "Resource not found"},
-    )