diff --git a/pipelines/pingcap/tiflash/latest/pull_integration_test.groovy b/pipelines/pingcap/tiflash/latest/pull_integration_test.groovy
index 49972a2e1..6cc3a5e76 100644
--- a/pipelines/pingcap/tiflash/latest/pull_integration_test.groovy
+++ b/pipelines/pingcap/tiflash/latest/pull_integration_test.groovy
@@ -360,7 +360,12 @@ pipeline {
                                     dir("tests/${TEST_PATH}") {
                                         echo "path: ${pwd()}"
                                         sh "docker ps -a && docker version"
-                                        sh "TAG=${tiflash_commit_hash} BRANCH=${REFS.base_ref} ./run.sh"
+                                        script {
+                                            def pdBranch = component.computeBranchFromPR('pd', REFS.base_ref, REFS.pulls[0].title, 'master')
+                                            def tikvBranch = component.computeBranchFromPR('tikv', REFS.base_ref, REFS.pulls[0].title, 'master')
+                                            def tidbBranch = component.computeBranchFromPR('tidb', REFS.base_ref, REFS.pulls[0].title, 'master')
+                                            sh "PD_BRANCH=${pdBranch} TIKV_BRANCH=${tikvBranch} TIDB_BRANCH=${tidbBranch} TAG=${tiflash_commit_hash} BRANCH=${REFS.base_ref} ./run.sh"
+                                        }
                                     }
                                 }
                             }