1 files changed, 3 insertions, 3 deletions
diff --git a/server/szurubooru/middleware/request_logger.py b/server/szurubooru/middleware/request_logger.py
index 47b43ab..54e40e4 100644
--- a/server/szurubooru/middleware/request_logger.py
+++ b/server/szurubooru/middleware/request_logger.py
@@ -1,5 +1,5 @@
 import logging
-from szurubooru import db
+from szurubooru import db, rest
 from szurubooru.rest import middleware
 
 
@@ -7,12 +7,12 @@ logger = logging.getLogger(__name__)
 
 
 @middleware.pre_hook
-def process_request(_ctx):
+def process_request(_ctx: rest.Context) -> None:
     db.reset_query_count()
 
 
 @middleware.post_hook
-def process_response(ctx):
+def process_response(ctx: rest.Context) -> None:
     logger.info(
         '%s %s (user=%s, queries=%d)',
         ctx.method,