diff --git a/modules/cloudbrain/cloudbrain.go b/modules/cloudbrain/cloudbrain.go
index 7d7d4a286..67f2aaf67 100755
--- a/modules/cloudbrain/cloudbrain.go
+++ b/modules/cloudbrain/cloudbrain.go
@@ -504,6 +504,7 @@ func RestartTask(ctx *context.Context, task *models.Cloudbrain, newID *string) e
 		JobType:         task.JobType,
 		Type:            task.Type,
 		Uuid:            task.Uuid,
+		DatasetName:     task.DatasetName,
 		Image:           task.Image,
 		GpuQueue:        task.GpuQueue,
 		ResourceSpecId:  task.ResourceSpecId,
diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index 597069aba..f9d45bcad 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -605,12 +605,6 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 		}
 
 	}
-	attachment, err := models.GetAttachmentByUUID(task.Uuid)
-	if err == nil {
-		ctx.Data["datasetname"] = attachment.Name
-	} else {
-		ctx.Data["datasetname"] = ""
-	}
 
 	ctx.Data["task"] = task
 	ctx.Data["jobName"] = task.JobName
diff --git a/templates/repo/cloudbrain/benchmark/show.tmpl b/templates/repo/cloudbrain/benchmark/show.tmpl
index c79f6f268..59ce3c471 100755
--- a/templates/repo/cloudbrain/benchmark/show.tmpl
+++ b/templates/repo/cloudbrain/benchmark/show.tmpl
@@ -474,7 +474,7 @@
 
                                                     <td class="ti-text-form-content">
                                                         <div class="text-span text-span-w" id="{{.VersionName}}-mirror">
-                                                            {{$.datasetname}}
+                                                            {{.DatasetName}}
                                                         </div>
                                                     </td>
                                                 </tr>
diff --git a/templates/repo/cloudbrain/show.tmpl b/templates/repo/cloudbrain/show.tmpl
index 34ddcf667..9ae3b9445 100755
--- a/templates/repo/cloudbrain/show.tmpl
+++ b/templates/repo/cloudbrain/show.tmpl
@@ -409,7 +409,7 @@
                                                     <td class="ti-text-form-content">
                                                         <div class="text-span text-span-w"
                                                             id="{{.VersionName}}-BenchmarkTypeName">
-                                                            {{$.datasetname}}
+                                                            {{.DatasetName}}
                                                         </div>
                                                     </td>
                                                 </tr>