diff --git a/.github/workflows/_linux_ut.yml b/.github/workflows/_linux_ut.yml
index 437c55c9e..edea430d2 100644
--- a/.github/workflows/_linux_ut.yml
+++ b/.github/workflows/_linux_ut.yml
@@ -58,7 +58,7 @@ jobs:
           source activate xpu_op_${ZE_AFFINITY_MASK}
           cd ../ && rm -rf pytorch
           git clone https://github.com/pytorch/pytorch pytorch
-          cd pytorch && git checkout $(echo "${{ inputs.pytorch }}" |awk '{print $1}') 
+          cd pytorch && git checkout $(echo "${{ inputs.pytorch }}" |sed 's/^nightly_wheel$/nightly/') 
           # apply PRs for stock pytorch
           pip install requests
           python ../torch-xpu-ops/.github/scripts/apply_torch_pr.py