diff --git a/.pipelines/ci/templates/run-ui-tests-ci.yml b/.pipelines/ci/templates/run-ui-tests-ci.yml
index 2ab4c0c342..b74212feae 100644
--- a/.pipelines/ci/templates/run-ui-tests-ci.yml
+++ b/.pipelines/ci/templates/run-ui-tests-ci.yml
@@ -21,6 +21,22 @@ jobs:
       artifact: build-${{ parameters.platform }}-${{ parameters.configuration }}
       path: $(Build.ArtifactStagingDirectory)
 
+  - task: UseDotNet@2
+    displayName: 'Use .NET 6 SDK'
+    inputs:
+      packageType: sdk
+      version: '6.x'
+
+  - task: UseDotNet@2
+    displayName: 'Use .NET 8 SDK'
+    inputs:
+      packageType: sdk
+      version: '8.x'
+      includePreviewVersions: true
+
+  - task: VisualStudioTestPlatformInstaller@1
+    displayName: Ensure VSTest Platform
+
   - task: VSTest@2
     displayName: 'UI Tests'
     condition: ne(variables['BuildPlatform'], 'arm64') # No arm64 agents to run the tests.
@@ -28,6 +44,7 @@ jobs:
       platform: '$(BuildPlatform)'
       configuration: '$(BuildConfiguration)'
       testSelector: 'testAssemblies'
+      uiTests: true
       searchFolder: '$(Build.ArtifactStagingDirectory)'
       testAssemblyVer2: |
         **\UITests-FancyZones.dll