diff --git a/.azure-pipelines/multi-nodes-test.yml b/.azure-pipelines/multi-nodes-test.yml
index 95dc5063..d92256ad 100644
--- a/.azure-pipelines/multi-nodes-test.yml
+++ b/.azure-pipelines/multi-nodes-test.yml
@@ -65,7 +65,7 @@ jobs:
     image: $(containerImage)
 
   steps:
-  - template: templates/sglang-test.yml
+  - template: templates/sglang-multi-test.yml
     parameters:
         subscription:  msccl-it
         vmssName:      mscclit-vmss
diff --git a/.azure-pipelines/templates/sglang-multi-test.yml b/.azure-pipelines/templates/sglang-multi-test.yml
index d4d4df3a..5373a2ff 100644
--- a/.azure-pipelines/templates/sglang-multi-test.yml
+++ b/.azure-pipelines/templates/sglang-multi-test.yml
@@ -6,8 +6,6 @@ parameters:
 - name: perfBaselineFile
   type: string
   default: 'test/deploy/perf_ndmv4.jsonl'
-- name: gpuArch
-  type: string
 - name: containerName
   type: string
   default: 'sglang-mscclpp-test'
@@ -30,7 +28,6 @@ steps:
   parameters:
     subscription:     ${{ parameters.subscription }}
     vmssName:         ${{ parameters.vmssName }}
-    gpuArch:          ${{ parameters.gpuArch }}
     deployArgs:       'single-node-test true cuda'
     containerName:    ${{ parameters.containerName }}