diff --git a/.pipelines/ci/templates/run-ui-tests-ci.yml b/.pipelines/ci/templates/run-ui-tests-ci.yml
index cb20dd502c..0884a3d62b 100644
--- a/.pipelines/ci/templates/run-ui-tests-ci.yml
+++ b/.pipelines/ci/templates/run-ui-tests-ci.yml
@@ -6,6 +6,8 @@ jobs:
 - job: UITest
   displayName: UI Test ${{ parameters.platform }} ${{ parameters.configuration }}
   dependsOn: Build${{ parameters.platform }}${{ parameters.configuration }}
+  variables:
+    SrcPath: $(Build.Repository.LocalPath)
   pool:
     ${{ if eq(variables['System.CollectionUri'], 'https://dev.azure.com/ms/') }}:
       name: SHINE-OSS-Testing-x64
diff --git a/src/modules/fancyzones/UITests-FancyZonesEditor/Init.cs b/src/modules/fancyzones/UITests-FancyZonesEditor/Init.cs
index 0e4eef5647..804c3324b3 100644
--- a/src/modules/fancyzones/UITests-FancyZonesEditor/Init.cs
+++ b/src/modules/fancyzones/UITests-FancyZonesEditor/Init.cs
@@ -17,9 +17,13 @@ namespace Microsoft.FancyZonesEditor.UITests
         [AssemblyInitialize]
         public static void SetupAll(TestContext context)
         {
-            string sourceDirPath = Path.GetFullPath($"{Environment.CurrentDirectory}" + @".\..\..\..\..\..\");
-            context.WriteLine($"source dir: {sourceDirPath}");
+            string? sourceDirPath = Environment.GetEnvironmentVariable("SrcPath"); // get source dir in CI
+            if (sourceDirPath == null)
+            {
+                sourceDirPath = Path.GetFullPath($"{Environment.CurrentDirectory}" + @".\..\..\..\..\..\"); // local
+            }
 
+            context.WriteLine($"source dir: {sourceDirPath}");
             string winAppDriver = Path.Combine(sourceDirPath, @".\deps\WinAppDriver", "WinAppDriver.exe");
 
             context.WriteLine($"Attempting to launch WinAppDriver at: {winAppDriver}");