diff --git a/.ci/azure-pipelines-v2.yml b/.ci/azure-pipelines-v2.yml
index 28f138a..a2b9b51 100644
--- a/.ci/azure-pipelines-v2.yml
+++ b/.ci/azure-pipelines-v2.yml
@@ -23,6 +23,7 @@ pr:
 stages:
 - template: .ci/stages/deploy_notebooks_stages_v2.yml@aitemplates
   parameters:
+    Agent: $(Agent_Name)
     jobDisplayName: MLScoreDeployJob
     DefaultWorkingDirectory: $(System.DefaultWorkingDirectory)
     workload_vars: ../vars/mlhyperparametertuning_vars.yml