diff --git a/azure-pipelines.yml b/azure-pipelines.yml
index b398725f..51827041 100644
--- a/azure-pipelines.yml
+++ b/azure-pipelines.yml
@@ -19,7 +19,7 @@ resources:
     - repository: microsoftai
       type: git
       name: one/AGAI-Industrial-CICD
-      ref: refs/heads/release/eap_alpha_5
+      ref: refs/heads/release/beta-1
     - repository: industrialai
       type: git
       name: one/AGAI-Industrial-AI
@@ -41,6 +41,7 @@ extends:
         variables: # More settings at https://aka.ms/obpipelines/yaml/jobs
           ob_outputDirectory: '$(Build.SourcesDirectory)/out' # this directory is uploaded to pipeline artifacts, reddog and cloudvault. More info at https://aka.ms/obpipelines/artifacts
         steps:
+        - checkout: self
         - checkout: industrialai
 
         - template: .ci/steps/python_ai_test.yml@microsoftai
@@ -53,3 +54,5 @@ extends:
           ob_outputDirectory: '$(Build.SourcesDirectory)/out' # this directory is uploaded to pipeline artifacts, reddog and cloudvault. More info at https://aka.ms/obpipelines/artifacts
         steps:
         - template: .ci/steps/python_ai_safety.yml@microsoftai
+          parameters:
+            SrcRoot: $(System.DefaultWorkingDirectory)/src