2 files changed, 8 insertions, 1 deletions

diff --git a/src/internal/SetBuildNumber/Directory.Packages.props.pp b/src/internal/SetBuildNumber/Directory.Packages.props.pp
index 8cf271d7..18a9b5eb 100644
--- a/src/internal/SetBuildNumber/Directory.Packages.props.pp
+++ b/src/internal/SetBuildNumber/Directory.Packages.props.pp
@@ -17,6 +17,7 @@
     <PackageVersion Include="WixToolset.Dtf.WindowsInstaller.Package" Version="{packageversion}" />
 
     <PackageVersion Include="WixInternal.TestSupport" Version="{packageversion}" />
+    <PackageVersion Include="WixInternal.MSTestSupport" Version="{packageversion}" />
     <PackageVersion Include="WixInternal.TestSupport.Native" Version="{packageversion}" />
     <PackageVersion Include="WixInternal.BaseBuildTasks.Sources" Version="{packageversion}" />
 
@@ -37,6 +38,7 @@
     <PackageVersion Include="WixToolset.Core.Burn" Version="{packageversion}" />
     <PackageVersion Include="WixToolset.Core.WindowsInstaller" Version="{packageversion}" />
     <PackageVersion Include="WixInternal.Core.TestPackage" Version="{packageversion}" />
+    <PackageVersion Include="WixInternal.Core.MSTestPackage" Version="{packageversion}" />
 
     <PackageVersion Include="WixToolset.Heat" Version="{packageversion}" />
 
@@ -98,6 +100,10 @@
   </ItemGroup>
 
   <ItemGroup>
+    <PackageVersion Include="MSTest.TestFramework" Version="3.7.3" />
+  </ItemGroup>
+
+  <ItemGroup>
     <PackageVersion Include="Microsoft.NET.Tools.NETCoreCheck.x86" Version="6.0.0" />
     <PackageVersion Include="Microsoft.NET.Tools.NETCoreCheck.x64" Version="6.0.0" />
     <PackageVersion Include="Microsoft.NET.Tools.NETCoreCheck.arm64" Version="6.0.0" />
diff --git a/src/internal/SetBuildNumber/global.json.pp b/src/internal/SetBuildNumber/global.json.pp
index bad3c0bd..fe48e6cc 100644
--- a/src/internal/SetBuildNumber/global.json.pp
+++ b/src/internal/SetBuildNumber/global.json.pp
@@ -1,6 +1,7 @@
 {
   "msbuild-sdks": {
-    "Microsoft.Build.Traversal": "3.2.0",
+    "MSTest.Sdk": "3.8.0",
+    "Microsoft.Build.Traversal": "4.1.82",
     "Microsoft.Build.NoTargets": "3.5.6",
     "WixToolset.Sdk": "{packageversion}"
   },