diff --git a/source/FluentDataBuilder.ExtensionTests/FluentDataBuilder.ExtensionTests.csproj b/source/FluentDataBuilder.ExtensionTests/FluentDataBuilder.ExtensionTests.csproj
index 510c4db..614a889 100644
--- a/source/FluentDataBuilder.ExtensionTests/FluentDataBuilder.ExtensionTests.csproj
+++ b/source/FluentDataBuilder.ExtensionTests/FluentDataBuilder.ExtensionTests.csproj
@@ -11,7 +11,7 @@
     <ItemGroup>
         <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.14.1" />
         <PackageReference Include="MSTest.TestAdapter" Version="3.10.2" />
-        <PackageReference Include="MSTest.TestFramework" Version="3.10.2" />
+        <PackageReference Include="MSTest.TestFramework" Version="4.0.1" />
         <PackageReference Include="coverlet.collector" Version="6.0.4">
             <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
             <PrivateAssets>all</PrivateAssets>