diff --git a/model/lib/extensions/presentation_mailbox_extension.dart b/model/lib/extensions/presentation_mailbox_extension.dart index 87a334ea8..a0f977725 100644 --- a/model/lib/extensions/presentation_mailbox_extension.dart +++ b/model/lib/extensions/presentation_mailbox_extension.dart @@ -53,7 +53,19 @@ extension PresentationMailboxExtension on PresentationMailbox { name?.name.toLowerCase() == PresentationMailbox.trashRole.toLowerCase(); } - bool get isDrafts => role == PresentationMailbox.roleDrafts; + bool get isDrafts { + if (isPersonal) { + return role == PresentationMailbox.roleDrafts; + } else { + return isDraftsTeamMailbox; + } + } + + bool get isDraftsTeamMailbox { + return isChildOfTeamMailboxes && + name?.name.toLowerCase() == + PresentationMailbox.draftsRole.toLowerCase(); + } bool get isTemplates { if (isPersonal) {