diff --git a/app/create_app.py b/app/create_app.py
index c449539..ad554f7 100644
--- a/app/create_app.py
+++ b/app/create_app.py
@@ -6,6 +6,7 @@ from starlette.middleware.cors import CORSMiddleware
 from starlette.responses import RedirectResponse
 from starlette.staticfiles import StaticFiles
 
+from app.middlewares.app_middlewares import add_app_middlewares
 from app.utils.db_utils import check_database_connection
 from app.utils.postgres_checkpointer import check_postgres_connection, close_postgres_connection
 
@@ -52,6 +53,8 @@ def create_app():
       redoc_js_url="/static/redoc.standalone.js",
     )
 
+  add_app_middlewares(app)
+
   app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
diff --git a/app/middlewares/app_middlewares.py b/app/middlewares/app_middlewares.py
index 7199936..b259574 100644
--- a/app/middlewares/app_middlewares.py
+++ b/app/middlewares/app_middlewares.py
@@ -64,7 +64,7 @@ def add_app_middlewares(app: FastAPI):
       response = await call_next(request)
     except HTTPException as e:
       if e.status_code == status.HTTP_401_UNAUTHORIZED:
-        return JSONResponse(content=e.detail, status_code=status.HTTP_401_UNAUTHORIZED)
+        return JSONResponse(content={"message": e.detail}, status_code=status.HTTP_401_UNAUTHORIZED)
       else:
         raise e
     return response
diff --git a/app/server.py b/app/server.py
index a065b28..08ecd13 100644
--- a/app/server.py
+++ b/app/server.py
@@ -18,23 +18,19 @@ from app.controller.test_sqlmodel import add_test_sqlmodel_route
 from app.controller.test_sync import add_test_sync_route
 from app.controller.translate_controller import add_translate_route
 from app.create_app import create_app
-from app.middlewares.app_middlewares import add_app_middlewares
 from app.model.LgApprove import LgApproveService
 from app.model.LgChat import LgChatService
 from app.model.LgMessage import LgMessageService
-from app.model.LlmOrder import LlmOrder, LlmOrderService
-from app.model.LlmProduct import LlmProduct, LlmProductService
+from app.model.LlmOrder import LlmOrderService
+from app.model.LlmProduct import LlmProductService
 from app.run_uvicorn import run_uvicorn
 from app.utils.ModelInputSchema import ModelInputSchema
 from app.utils.add_async_route import add_async_route
-from app.utils.create_module_service import create_model_service
 from app.utils.llm_utils import create_llm
 from app.utils.next_id import add_next_id_route
 
 app = create_app()
 
-add_app_middlewares(app)
-
 add_translate_route(app)
 add_custom_chat_playground_route(app)
 add_test_sync_route(app)