diff --git a/build-tools/automation/yaml-templates/apk-instrumentation.yaml b/build-tools/automation/yaml-templates/apk-instrumentation.yaml
index 20286848482..7f8b03133bf 100644
--- a/build-tools/automation/yaml-templates/apk-instrumentation.yaml
+++ b/build-tools/automation/yaml-templates/apk-instrumentation.yaml
@@ -11,6 +11,7 @@ parameters:
   artifactFolder: ""
   useDotNet: true
   condition: succeeded()
+  retryCountOnTaskFailure: 1
 
 steps:
 - ${{ if eq(parameters.useDotNet, false) }}:
@@ -26,6 +27,7 @@ steps:
         ${{ parameters.extraBuildArgs }}
     condition: ${{ parameters.condition }}
     continueOnError: true
+    retryCountOnTaskFailure: ${{ parameters.retryCountOnTaskFailure }}
 
 - ${{ if eq(parameters.useDotNet, true) }}:
   - template: run-dotnet-preview.yaml
@@ -40,6 +42,7 @@ steps:
         -v:n -c ${{ parameters.configuration }} ${{ parameters.extraBuildArgs }}
       condition: ${{ parameters.condition }}
       continueOnError: true
+      retryCountOnTaskFailure: ${{ parameters.retryCountOnTaskFailure }}
 
 - script: >
     DEST="$(Build.StagingDirectory)/Test${{ parameters.configuration }}/${{ parameters.artifactFolder }}/" &&
diff --git a/build-tools/automation/yaml-templates/run-dotnet-preview.yaml b/build-tools/automation/yaml-templates/run-dotnet-preview.yaml
index cad0476d48b..f0b62b4eb1d 100644
--- a/build-tools/automation/yaml-templates/run-dotnet-preview.yaml
+++ b/build-tools/automation/yaml-templates/run-dotnet-preview.yaml
@@ -8,6 +8,7 @@ parameters:
   displayName: run dotnet preview
   condition: succeeded()
   continueOnError: true
+  retryCountOnTaskFailure: 0
 
 steps:
 - powershell: |
@@ -25,3 +26,4 @@ steps:
   displayName: ${{ parameters.displayName }}
   condition: ${{ parameters.condition }}
   continueOnError: ${{ parameters.continueOnError }}
+  retryCountOnTaskFailure: ${{ parameters.retryCountOnTaskFailure }}