From 2f215cb24b101113c3757f1df001e39648f4b464 Mon Sep 17 00:00:00 2001
From: chynesNR <chynes@newrelic.com>
Date: Wed, 4 Sep 2024 15:34:32 -0700
Subject: [PATCH] Missed a spot

---
 .../Agent/IntegrationTests/IntegrationTests/LLM/BedrockTests.cs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/Agent/IntegrationTests/IntegrationTests/LLM/BedrockTests.cs b/tests/Agent/IntegrationTests/IntegrationTests/LLM/BedrockTests.cs
index 623a1acd01..720e9033cf 100644
--- a/tests/Agent/IntegrationTests/IntegrationTests/LLM/BedrockTests.cs
+++ b/tests/Agent/IntegrationTests/IntegrationTests/LLM/BedrockTests.cs
@@ -112,7 +112,7 @@ public void BedrockTest()
         {
             var expectedMetrics = new List<Assertions.ExpectedMetric>
             {
-                new Assertions.ExpectedMetric { metricName = @"Custom/Llm/completion/Bedrock/InvokeModelAsync", CallCountAllHarvests = 5 },
+                new Assertions.ExpectedMetric { metricName = @"Custom/Llm/completion/Bedrock/InvokeModelAsync", CallCountAllHarvests = _bedrockModelsToTest.Count - 1 },
                 new Assertions.ExpectedMetric { metricName = @"Custom/Llm/embedding/Bedrock/InvokeModelAsync", CallCountAllHarvests = 1 },
                 new Assertions.ExpectedMetric { metricName = @"Supportability/DotNet/ML/.*", IsRegexName = true}
             };