diff --git a/src/Messaging/NBB.Messaging.Host/Internal/MessagingHost.cs b/src/Messaging/NBB.Messaging.Host/Internal/MessagingHost.cs
index b9b32eb3..a2dc303a 100644
--- a/src/Messaging/NBB.Messaging.Host/Internal/MessagingHost.cs
+++ b/src/Messaging/NBB.Messaging.Host/Internal/MessagingHost.cs
@@ -60,15 +60,15 @@ public void ScheduleRestart(TimeSpan delay = default)
                 return;
 
             _logger.LogInformation($"Messaging host is scheduled for restart in {delay.TotalSeconds} seconds");
+           
+            using IDisposable _ = ExecutionContext.IsFlowSuppressed() ? null : ExecutionContext.SuppressFlow();
+
             Task.Run(async () =>
             {
                 try
                 {
                     await Task.Delay(delay);
 
-                    if (!ExecutionContext.IsFlowSuppressed())
-                        ExecutionContext.SuppressFlow();
-
                     await TryStopAsync();
                     await StartAsync();
                 }