diff --git a/src/ez_lan_manager/helpers/LoggedInGuard.py b/src/ez_lan_manager/helpers/LoggedInGuard.py
index 04aa8ca..0e50427 100644
--- a/src/ez_lan_manager/helpers/LoggedInGuard.py
+++ b/src/ez_lan_manager/helpers/LoggedInGuard.py
@@ -1,16 +1,16 @@
 from typing import Optional
 
-from rio import Session, URL
+from rio import Session, URL, GuardEvent
 
 from src.ez_lan_manager.types.SessionStorage import SessionStorage
 
 
 # Guards pages against access from users that are NOT logged in
-def logged_in_guard(session: Session, _) -> Optional[URL]:
-    if session[SessionStorage].user_id is None:
+def logged_in_guard(event: GuardEvent) -> Optional[URL]:
+    if event.session[SessionStorage].user_id is None:
         return URL("./")
 
 # Guards pages against access from users that ARE logged in
-def not_logged_in_guard(session: Session, _) -> Optional[URL]:
-    if session[SessionStorage].user_id is not None:
+def not_logged_in_guard(event: GuardEvent) -> Optional[URL]:
+    if event.session[SessionStorage].user_id is not None:
         return URL("./")