using cosmosexception

Azure · Dec 16, 2020 · 0b462bd · 0b462bd
1 parent 337d0dc
commit 0b462bd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Microsoft.Azure.Cosmos/src/BulkPartitionKeyRangeGoneRetryPolicy.cs b/Microsoft.Azure.Cosmos/src/BulkPartitionKeyRangeGoneRetryPolicy.cs
@@ -33,11 +33,11 @@ internal sealed class BulkPartitionKeyRangeGoneRetryPolicy : IDocumentClientRetr
             Exception exception,
             CancellationToken cancellationToken)
         {
-            if (exception is DocumentClientException clientException)
+            if (exception is CosmosException clientException)
             {
                 ShouldRetryResult shouldRetryResult = await this.ShouldRetryInternalAsync(
-                    clientException?.StatusCode,
-                    clientException?.GetSubStatus(),
+                    clientException.StatusCode,
+                    (SubStatusCodes)clientException.SubStatusCode,
                     cancellationToken);
 
                 if (shouldRetryResult != null)