diff --git a/tools/ci_build/github/azure-pipelines/templates/c-api-linux-cpu.yml b/tools/ci_build/github/azure-pipelines/templates/c-api-linux-cpu.yml
index c0175acc5a..17a7dd8de2 100644
--- a/tools/ci_build/github/azure-pipelines/templates/c-api-linux-cpu.yml
+++ b/tools/ci_build/github/azure-pipelines/templates/c-api-linux-cpu.yml
@@ -39,10 +39,7 @@ jobs:
   pool:
     name: ${{parameters.PoolName}}
     os: linux
-    ${{ if eq(parameters.OnnxruntimeArch, 'aarch64') }}:
-      hostArchitecture: Arm64
-    ${{ if eq(parameters.OnnxruntimeArch, 'x64') }}:
-      hostArchitecture: x64
+    hostArchitecture: ${{parameters.OnnxruntimeNodejsBindingArch}}
   steps:
   - checkout: self
     clean: true