diff --git a/main.py b/main.py
index 06763c1..3f7f8f8 100644
--- a/main.py
+++ b/main.py
@@ -30,7 +30,8 @@ async def notify_start():
 
 
 async def main():
-    dp.update.outer_middleware(AuditMiddleware(cfg))
+    dp.message.middleware(AuditMiddleware(cfg))
+    dp.callback_query.middleware(AuditMiddleware(cfg))
     audit_start(cfg)
     state.DOCKER_MAP.clear()
     state.DOCKER_MAP.update(await discover_containers(cfg))