diff --git a/Microsoft.Azure.Cosmos/src/BulkPartitionKeyRangeGoneRetryPolicy.cs b/Microsoft.Azure.Cosmos/src/BulkPartitionKeyRangeGoneRetryPolicy.cs
index 3bf4bf8cbde..0191c13e522 100644
--- 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)