diff --git a/sdk/batch/Microsoft.Azure.Batch/tests/IntegrationTests/BatchRequestIntegrationTests.cs b/sdk/batch/Microsoft.Azure.Batch/tests/IntegrationTests/BatchRequestIntegrationTests.cs
index 07982f522327..56c5b549db34 100644
--- a/sdk/batch/Microsoft.Azure.Batch/tests/IntegrationTests/BatchRequestIntegrationTests.cs
+++ b/sdk/batch/Microsoft.Azure.Batch/tests/IntegrationTests/BatchRequestIntegrationTests.cs
@@ -111,7 +111,7 @@ await SynchronizationContextHelper.RunTestAsync(async () =>
                             req.CancellationToken = tokenSource.Token;
                         }));
 
-                        await TestUtilities.AssertThrowsAsync<System.Threading.Tasks.TaskCanceledException>(async () =>
+                        await TestUtilities.AssertThrowsAsync<OperationCanceledException>(async () =>
                             await client.JobOperations.GetJobAsync("Foo", additionalBehaviors: customBehaviors).ConfigureAwait(false)).ConfigureAwait(false);
                     }
                 }