diff --git a/backend/src/main/java/io/dataease/service/dataset/DataSetTableService.java b/backend/src/main/java/io/dataease/service/dataset/DataSetTableService.java
index 2753db3567..4c639c23d1 100644
--- a/backend/src/main/java/io/dataease/service/dataset/DataSetTableService.java
+++ b/backend/src/main/java/io/dataease/service/dataset/DataSetTableService.java
@@ -102,7 +102,7 @@ public class DataSetTableService {
             int update = datasetTableMapper.updateByPrimaryKeySelective(datasetTable);
             // sql 更新
             if (update == 1) {
-                if (StringUtils.equalsIgnoreCase(datasetTable.getType(), "sql")) {
+                if (StringUtils.equalsIgnoreCase(datasetTable.getType(), "sql") || StringUtils.equalsIgnoreCase(datasetTable.getType(), "custom")) {
                     // 删除所有字段，重新抽象
                     dataSetTableFieldsService.deleteByTableId(datasetTable.getId());
                     saveTableField(datasetTable);
diff --git a/frontend/src/views/dataset/add/AddCustom.vue b/frontend/src/views/dataset/add/AddCustom.vue
index ddc55358fb..2a8fc2a3d9 100644
--- a/frontend/src/views/dataset/add/AddCustom.vue
+++ b/frontend/src/views/dataset/add/AddCustom.vue
@@ -23,7 +23,7 @@
     </el-row>
     <el-col style="display: flex;flex-direction: row">
       <el-col class="panel-height" style="width: 220px;border-right:solid 1px #dcdfe6;border-top:solid 1px #dcdfe6;padding-right: 15px;overflow-y: auto;">
-        <dataset-group-selector :mode="1" :checked-list="checkedList" :union-data="unionData" @getTable="getTable" />
+        <dataset-group-selector :table="table" :mode="1" :checked-list="checkedList" :union-data="unionData" @getTable="getTable" />
       </el-col>
       <el-col class="panel-height" style="width: 235px;border-top:solid 1px #dcdfe6;padding: 0 15px;overflow-y: auto;">
         <dataset-custom-field :table="table" :checked-list="checkedList" @getChecked="getChecked" />
diff --git a/frontend/src/views/dataset/common/DatasetGroupSelector.vue b/frontend/src/views/dataset/common/DatasetGroupSelector.vue
index 4e160b70c1..d3ce447410 100644
--- a/frontend/src/views/dataset/common/DatasetGroupSelector.vue
+++ b/frontend/src/views/dataset/common/DatasetGroupSelector.vue
@@ -122,6 +122,11 @@ export default {
       type: Array,
       required: false,
       default: null
+    },
+    table: {
+      type: Object,
+      required: false,
+      default: null
     }
   },
   data() {
@@ -153,6 +158,18 @@ export default {
     'unionData': function() {
       this.unionDataChange()
     },
+    'table': function() {
+      if (this.table && this.table.sceneId) {
+        post('dataset/group/getScene/' + this.table.sceneId, {}).then(response => {
+          this.currGroup = response.data
+
+          this.$nextTick(function() {
+            this.sceneMode = true
+            this.tableTree()
+          })
+        })
+      }
+    },
     search(val) {
       if (val && val !== '') {
         this.tableData = JSON.parse(JSON.stringify(this.tables.filter(ele => { return ele.name.includes(val) })))
@@ -165,10 +182,6 @@ export default {
     this.tree(this.groupForm)
     this.tableTree()
   },
-  activated() {
-    this.tree(this.groupForm)
-    this.tableTree()
-  },
   created() {
     this.kettleState()
   },