diff --git a/SW.EfCoreExtensions.PgSql.IntegrationTests/SW.EfCoreExtensions.PgSql.IntegrationTests.csproj b/SW.EfCoreExtensions.PgSql.IntegrationTests/SW.EfCoreExtensions.PgSql.IntegrationTests.csproj
index bd62603..e345507 100644
--- a/SW.EfCoreExtensions.PgSql.IntegrationTests/SW.EfCoreExtensions.PgSql.IntegrationTests.csproj
+++ b/SW.EfCoreExtensions.PgSql.IntegrationTests/SW.EfCoreExtensions.PgSql.IntegrationTests.csproj
@@ -8,7 +8,7 @@
   <ItemGroup>
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.12.0" />
     <PackageReference Include="MSTest.TestAdapter" Version="3.7.1" />
-    <PackageReference Include="MSTest.TestFramework" Version="3.7.1" />
+    <PackageReference Include="MSTest.TestFramework" Version="4.5.2" />
     <PackageReference Include="Testcontainers.PostgreSql" Version="3.10.0" />
   </ItemGroup>
 
diff --git a/SW.EfCoreExtensions.UnitTests/SW.EfCoreExtensions.UnitTests.csproj b/SW.EfCoreExtensions.UnitTests/SW.EfCoreExtensions.UnitTests.csproj
index cc84df8..7a0d61e 100644
--- a/SW.EfCoreExtensions.UnitTests/SW.EfCoreExtensions.UnitTests.csproj
+++ b/SW.EfCoreExtensions.UnitTests/SW.EfCoreExtensions.UnitTests.csproj
@@ -10,7 +10,7 @@
     <PackageReference Include="Microsoft.AspNetCore.Mvc.Testing" Version="10.0.12" />
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.12.0" />
     <PackageReference Include="MSTest.TestAdapter" Version="3.7.1" />
-    <PackageReference Include="MSTest.TestFramework" Version="3.7.1" />
+    <PackageReference Include="MSTest.TestFramework" Version="4.5.2" />
     <PackageReference Include="Bogus" Version="35.6.1" />
     <PackageReference Include="Microsoft.EntityFrameworkCore.Sqlite" Version="10.0.12" />