2 files changed, 3 insertions, 2 deletions

diff --git a/src/internal/SetBuildNumber/Directory.Packages.props.pp b/src/internal/SetBuildNumber/Directory.Packages.props.pp
index ff73265c..4743fb2e 100644
--- a/src/internal/SetBuildNumber/Directory.Packages.props.pp
+++ b/src/internal/SetBuildNumber/Directory.Packages.props.pp
@@ -50,11 +50,11 @@
     <PackageVersion Include="NuGet.Versioning" Version="5.6.0" />
   </ItemGroup>
 
-  <ItemGroup Condition="'$(TargetFramework)'=='net461' or '$(TargetFramework)'=='net472'" >
+  <ItemGroup Condition="'$(TargetFrameworkIdentifier)' == '.NETFramework'" >
     <PackageVersion Include="Microsoft.Build.Tasks.Core" Version="14.3"/>
   </ItemGroup>
 
-  <ItemGroup Condition="'$(TargetFramework)'=='netstandard2.0' or '$(TargetFramework)'=='netcoreapp3.1'">
+  <ItemGroup Condition="'$(TargetFrameworkIdentifier)' != '.NETFramework'">
     <PackageVersion Include="Microsoft.Build.Tasks.Core" Version="15.7.179" />
   </ItemGroup>
 
diff --git a/src/internal/SetBuildNumber/global.json.pp b/src/internal/SetBuildNumber/global.json.pp
index a1d6244a..5907efb0 100644
--- a/src/internal/SetBuildNumber/global.json.pp
+++ b/src/internal/SetBuildNumber/global.json.pp
@@ -1,5 +1,6 @@
 {
   "msbuild-sdks": {
+    "Microsoft.Build.Traversal": "3.0.23",
     "WixToolset.Sdk": "{packageversion}"
   },
   "sdk": {