diff --git a/src/graphql_schema/entities/flight.py b/src/graphql_schema/entities/flight.py
index ac08e75..6c9bab7 100644
--- a/src/graphql_schema/entities/flight.py
+++ b/src/graphql_schema/entities/flight.py
@@ -62,4 +62,4 @@ class FlightMutation:
 
     @strawberry.mutation(extensions=[AuthenticatedOnly()])
     async def delete_flight(self, info, id: int) -> Flight:
-        return await FlightMutationResolver().delete(info.context.user_id, id)
+        return await FlightMutationResolver().delete(info.context, id)
diff --git a/src/jwt.py b/src/jwt.py
index 17199df..5de164d 100644
--- a/src/jwt.py
+++ b/src/jwt.py
@@ -1,6 +1,6 @@
 from datetime import timedelta
 from fastapi_jwt import JwtAccessBearerCookie, JwtRefreshBearerCookie
-from .config import APP_SECRET_KEY, APP_DEBUG, REFRESH_TOKEN_VALIDITY_DAYS
+from config import APP_SECRET_KEY, APP_DEBUG, REFRESH_TOKEN_VALIDITY_DAYS
 
 access_security = JwtAccessBearerCookie(
     secret_key=APP_SECRET_KEY,