diff --git a/src/tests/Cohere.IntegrationTests/Cohere.IntegrationTests.csproj b/src/tests/Cohere.IntegrationTests/Cohere.IntegrationTests.csproj
index e351dd2..0ea2962 100644
--- a/src/tests/Cohere.IntegrationTests/Cohere.IntegrationTests.csproj
+++ b/src/tests/Cohere.IntegrationTests/Cohere.IntegrationTests.csproj
@@ -10,8 +10,8 @@
       <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
     </PackageReference>
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.9.0" />
-    <PackageReference Include="MSTest.TestAdapter" Version="3.2.2" />
-    <PackageReference Include="MSTest.TestFramework" Version="3.2.2" />
+    <PackageReference Include="MSTest.TestAdapter" Version="3.3.1" />
+    <PackageReference Include="MSTest.TestFramework" Version="3.3.1" />
     <PackageReference Include="FluentAssertions" Version="6.12.0" />
   </ItemGroup>