diff --git a/lib/features/email/presentation/controller/single_email_controller.dart b/lib/features/email/presentation/controller/single_email_controller.dart index 1efd7cae9..10cf0e792 100644 --- a/lib/features/email/presentation/controller/single_email_controller.dart +++ b/lib/features/email/presentation/controller/single_email_controller.dart @@ -344,8 +344,8 @@ class SingleEmailController extends BaseController with AppLoaderMixin { ); } else if (action is RemoveLabelFromEmailAction) { mailboxDashBoardController.clearEmailUIAction(); - if (_currentEmailId == null || - action.emailId != _currentEmailId) { + final displayedEmailId = _currentEmailId ?? currentEmail?.id; + if (displayedEmailId == null || action.emailId != displayedEmailId) { return; } mailboxDashBoardController.toggleLabelToEmail( @@ -355,8 +355,8 @@ class SingleEmailController extends BaseController with AppLoaderMixin { ); } else if (action is SyncUpdateLabelForEmailOnMemory) { mailboxDashBoardController.clearEmailUIAction(); - if (_currentEmailId == null || - action.emailId != _currentEmailId) { + final displayedEmailId = _currentEmailId ?? currentEmail?.id; + if (displayedEmailId == null || action.emailId != displayedEmailId) { return; }