From 1b9bc5142d15afd4a7c960f1103ea4b91ccf136b Mon Sep 17 00:00:00 2001
From: Mike Pateras <mipatera@microsoft.com>
Date: Wed, 30 Sep 2020 14:24:40 -0400
Subject: [PATCH] Fixed integration test.

---
 .../tests/IntegrationTests/BatchRequestIntegrationTests.cs      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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);
                     }
                 }