diff --git a/dotnet/tests/Microsoft.Agents.Workflows.Declarative.IntegrationTests/AzureAgentProviderTest.cs b/dotnet/tests/Microsoft.Agents.Workflows.Declarative.IntegrationTests/AzureAgentProviderTest.cs
index 2412937e48..a66240990b 100644
--- a/dotnet/tests/Microsoft.Agents.Workflows.Declarative.IntegrationTests/AzureAgentProviderTest.cs
+++ b/dotnet/tests/Microsoft.Agents.Workflows.Declarative.IntegrationTests/AzureAgentProviderTest.cs
@@ -3,6 +3,7 @@
 using System;
 using System.Linq;
 using System.Threading.Tasks;
+using Azure;
 using Azure.AI.Agents.Persistent;
 using Azure.Identity;
 using Microsoft.Agents.AI;
@@ -69,7 +70,7 @@ public sealed class AzureAgentProviderTest(ITestOutputHelper output) : WorkflowT
         Assert.Equal(agent2Id, agent2.Id);
 
         // Act & Assert
-        await Assert.ThrowsAsync<DeclarativeActionException>(() => provider.GetAgentAsync(agentName));
+        await Assert.ThrowsAsync<RequestFailedException>(() => provider.GetAgentAsync(agentName));
     }
 
     private async ValueTask<string> CreateAgentAsync(string? name = null)