diff --git a/plugin/migration/api.go b/plugin/migration/api.go
index 908727f8..5060c9bc 100644
--- a/plugin/migration/api.go
+++ b/plugin/migration/api.go
@@ -75,6 +75,10 @@ func (h *APIHandler) createDataMigrationTask(w http.ResponseWriter, req *http.Re
 		totalDocs += index.Source.Docs
 	}
 
+	srcClusterCfg := elastic.GetConfig(clusterTaskConfig.Cluster.Source.Id)
+	clusterTaskConfig.Cluster.Source.Distribution = srcClusterCfg.Distribution
+	dstClusterCfg := elastic.GetConfig(clusterTaskConfig.Cluster.Target.Id)
+	clusterTaskConfig.Cluster.Target.Distribution = dstClusterCfg.Distribution
 	t := task2.Task{
 		Metadata: task2.Metadata{
 			Type: "pipeline",
diff --git a/plugin/migration/model.go b/plugin/migration/model.go
index 9baf15a4..8e41952f 100644
--- a/plugin/migration/model.go
+++ b/plugin/migration/model.go
@@ -84,6 +84,7 @@ func (ii *IndexInfo) GetUniqueIndexName() string{
 type ClusterInfo struct {
 	Id   string `json:"id"`
 	Name string `json:"name"`
+	Distribution string `json:"distribution,omitempty"`
 }
 
 type TaskCompleteState struct {