diff --git a/authentik/tasks/models.py b/authentik/tasks/models.py
index 9867cb895c..3648e45c21 100644
--- a/authentik/tasks/models.py
+++ b/authentik/tasks/models.py
@@ -6,6 +6,7 @@ import pgtrigger
 from django.contrib.contenttypes.fields import GenericForeignKey
 from django.contrib.contenttypes.models import ContentType
 from django.db import models
+from django.db.models import QuerySet
 from django.utils.translation import gettext_lazy as _
 from django_dramatiq_postgres.models import TaskBase, TaskState
 from dramatiq.errors import Retry