diff --git a/src/oidc-test/main.py b/src/oidc-test/main.py
index 58d7b50..46e6594 100644
--- a/src/oidc-test/main.py
+++ b/src/oidc-test/main.py
@@ -204,7 +204,7 @@ async def logout(request: Request) -> RedirectResponse:
     return RedirectResponse(url="/")
 
 
-@app.get("/", response_class=HTMLResponse)
+@app.get("/")
 async def home(
     request: Request, user: Annotated[User, Depends(get_current_user_or_none)]
 ) -> HTMLResponse:
@@ -220,5 +220,7 @@ async def home(
 
 
 @app.get("/protected")
-async def get_protected(user: Annotated[User, Depends(get_current_user)]):
+async def get_protected(
+    user: Annotated[User, Depends(get_current_user)]
+) -> PlainTextResponse:
     return PlainTextResponse("Only authenticated users can see this")