diff --git a/.pipelines/ci/templates/run-ui-tests-ci.yml b/.pipelines/ci/templates/run-ui-tests-ci.yml
index 076d950263..2ab4c0c342 100644
--- a/.pipelines/ci/templates/run-ui-tests-ci.yml
+++ b/.pipelines/ci/templates/run-ui-tests-ci.yml
@@ -19,6 +19,7 @@ jobs:
     displayName: Download artifacts
     inputs:
       artifact: build-${{ parameters.platform }}-${{ parameters.configuration }}
+      path: $(Build.ArtifactStagingDirectory)
 
   - task: VSTest@2
     displayName: 'UI Tests'
@@ -27,6 +28,7 @@ jobs:
       platform: '$(BuildPlatform)'
       configuration: '$(BuildConfiguration)'
       testSelector: 'testAssemblies'
+      searchFolder: '$(Build.ArtifactStagingDirectory)'
       testAssemblyVer2: |
         **\UITests-FancyZones.dll
         **\UITests-FancyZonesEditor.dll