diff --git a/frontend/src/views/dataset/add/AddSQL.vue b/frontend/src/views/dataset/add/AddSQL.vue
index 25c7b80365..a5860f5e15 100644
--- a/frontend/src/views/dataset/add/AddSQL.vue
+++ b/frontend/src/views/dataset/add/AddSQL.vue
@@ -911,8 +911,9 @@ export default {
     },
 
     initTableInfo() {
-      if (this.param.tableId) {
-        getTable(this.param.tableId).then((response) => {
+      const tableId = this.param.tableId || this.$route.query.id
+      if (tableId) {
+        getTable(tableId).then((response) => {
           const table = response.data
           this.dataSource = table.dataSourceId
           this.changeDatasource()