diff --git a/authentik/core/tests/test_users_api.py b/authentik/core/tests/test_users_api.py
index 60f48cc3cb..773b5006c1 100644
--- a/authentik/core/tests/test_users_api.py
+++ b/authentik/core/tests/test_users_api.py
@@ -3,7 +3,6 @@
 from datetime import datetime, timedelta
 from json import loads
 
-from django.conf import settings
 from django.contrib.auth.hashers import make_password
 from django.urls.base import reverse
 from django.utils.timezone import now
@@ -12,7 +11,9 @@ from rest_framework.test import APITestCase
 from authentik.brands.models import Brand
 from authentik.core.models import (
     USER_ATTRIBUTE_TOKEN_EXPIRING,
+    AuthenticatedSession,
     Group,
+    Session,
     Token,
     User,
     UserTypes,
diff --git a/authentik/providers/oauth2/tests/test_authorize.py b/authentik/providers/oauth2/tests/test_authorize.py
index 18dc50a3f4..0ff4b170c0 100644
--- a/authentik/providers/oauth2/tests/test_authorize.py
+++ b/authentik/providers/oauth2/tests/test_authorize.py
@@ -10,7 +10,6 @@ from django.utils.timezone import now
 
 from authentik.blueprints.tests import apply_blueprint
 from authentik.common.oauth.constants import (
-    QS_LOGIN_HINT,
     SCOPE_OFFLINE_ACCESS,
     SCOPE_OPENID,
     TOKEN_TYPE,
@@ -18,7 +17,7 @@ from authentik.common.oauth.constants import (
 from authentik.core.models import Application
 from authentik.core.tests.utils import create_test_admin_user, create_test_brand, create_test_flow
 from authentik.events.models import Event, EventAction
-from authentik.flows.models import FlowDesignation, FlowStageBinding
+from authentik.flows.models import FlowStageBinding
 from authentik.flows.stage import PLAN_CONTEXT_PENDING_USER_IDENTIFIER
 from authentik.flows.views.executor import QS_QUERY, SESSION_KEY_PLAN
 from authentik.lib.generators import generate_id