diff --git a/src/documents/admin.py b/src/documents/admin.py
index 4c034829a..5df235618 100644
--- a/src/documents/admin.py
+++ b/src/documents/admin.py
@@ -108,7 +108,7 @@ class SavedViewAdmin(GuardedModelAdmin):
 
     inlines = [RuleInline]
 
-    def get_queryset(self, request):
+    def get_queryset(self, request):  # pragma: no cover
         return super().get_queryset(request).select_related("owner")
 
 
@@ -145,7 +145,7 @@ class NotesAdmin(GuardedModelAdmin):
     raw_id_fields = ("document",)
     search_fields = ("document__title",)
 
-    def get_queryset(self, request):
+    def get_queryset(self, request):  # pragma: no cover
         return (
             super()
             .get_queryset(request)
@@ -159,7 +159,7 @@ class ShareLinksAdmin(GuardedModelAdmin):
     list_display_links = ("created",)
     raw_id_fields = ("document",)
 
-    def get_queryset(self, request):
+    def get_queryset(self, request):  # pragma: no cover
         return super().get_queryset(request).select_related("document__correspondent")
 
 
@@ -177,7 +177,7 @@ class CustomFieldInstancesAdmin(GuardedModelAdmin):
     search_fields = ("document__title",)
     list_filter = ("created", "field")
 
-    def get_queryset(self, request):
+    def get_queryset(self, request):  # pragma: no cover
         return (
             super()
             .get_queryset(request)
diff --git a/src/paperless_mail/admin.py b/src/paperless_mail/admin.py
index ed7848381..be9df4c44 100644
--- a/src/paperless_mail/admin.py
+++ b/src/paperless_mail/admin.py
@@ -119,7 +119,7 @@ class MailRuleAdmin(GuardedModelAdmin):
 
     ordering = ["order"]
 
-    raw_id_fields = ("assign_correspondent",)
+    raw_id_fields = ("assign_correspondent", "assign_document_type")
 
     filter_horizontal = ("assign_tags",)