diff --git a/sdk/servicebus/Azure.Messaging.ServiceBus/src/ServiceBusRetryPolicy.cs b/sdk/servicebus/Azure.Messaging.ServiceBus/src/ServiceBusRetryPolicy.cs index 5690dbd7a30d..27e242ac0033 100644 --- a/sdk/servicebus/Azure.Messaging.ServiceBus/src/ServiceBusRetryPolicy.cs +++ b/sdk/servicebus/Azure.Messaging.ServiceBus/src/ServiceBusRetryPolicy.cs @@ -119,7 +119,7 @@ internal async Task RunOperation( { // We are in a server busy state before we start processing. // Since ServerBusyBaseSleepTime > remaining time for the operation, we don't wait for the entire Sleep time. - await Task.Delay(tryTimeout).ConfigureAwait(false); + await Task.Delay(tryTimeout, cancellationToken).ConfigureAwait(false); throw new ServiceBusException( ServerBusyExceptionMessage, ServiceBusException.FailureReason.ServiceBusy); @@ -128,7 +128,7 @@ internal async Task RunOperation( { if (IsServerBusy) { - await Task.Delay(ServerBusyBaseSleepTime).ConfigureAwait(false); + await Task.Delay(ServerBusyBaseSleepTime, cancellationToken).ConfigureAwait(false); } try