diff --git a/core/frontend/src/views/dataset/add/AddSQL.vue b/core/frontend/src/views/dataset/add/AddSQL.vue
index 6c91d75b69..22eb2b493c 100644
--- a/core/frontend/src/views/dataset/add/AddSQL.vue
+++ b/core/frontend/src/views/dataset/add/AddSQL.vue
@@ -966,6 +966,10 @@ export default {
               table.info.replace(/\n/g, '\\n').replace(/\r/g, '\\r')
             ).sql
           }
+          if(JSON.parse(table.info).hasOwnProperty("setKey")){
+            this.param.setKey = JSON.parse(table.info).setKey
+            this.param.keys = JSON.parse(table.info).keys
+          }
           this.variables = JSON.parse(table.sqlVariableDetails)
         }).finally(() => {
           this.$emit('setSaveDisabled', false)