diff --git a/aleksis/apps/alsijil/util/alsijil_helpers.py b/aleksis/apps/alsijil/util/alsijil_helpers.py
index f44a90944d8b1e8d6360012b7b818a2f77fd4359..0ea834badf5b646c971ce437a38778021c9a7328 100644
--- a/aleksis/apps/alsijil/util/alsijil_helpers.py
+++ b/aleksis/apps/alsijil/util/alsijil_helpers.py
@@ -76,12 +76,12 @@ def annotate_documentations(
         instances = instances.order_by("period__weekday", "period__period")
     else:
         instances = instances.order_by("period_from__weekday", "period_from__period")
-    args = {Event: "event", LessonPeriod: "lesson_period", ExtraLesson: "extra_lesson"}
+
     instances = instances.annotate(
         has_documentation=Exists(
             LessonDocumentation.objects.filter(
                 ~Q(topic__exact=""), week=wanted_week.week, year=wanted_week.year,
-            ).filter(**{args[klass]: OuterRef("pk")})
+            ).filter(**{klass.label_: OuterRef("pk")})
         )
     )