diff --git a/web_src/js/index.js b/web_src/js/index.js
index b7434885e..b9e7165cf 100755
--- a/web_src/js/index.js
+++ b/web_src/js/index.js
@@ -3923,7 +3923,7 @@ function initVueDataset() {
       this.getTypeList()
 
       if (!!document.getElementById('dataset-repolink-init')) {
-        this.cloudbrainType = location.href.indexOf('cloudbrain/train-job/create') !== -1 ? 0 : 1
+        this.cloudbrainType = location.href.indexOf('cloudbrain') !== -1 ? 0 : 1
         this.getCurrentRepoDataset(this.repolink, this.cloudbrainType)
       }