diff --git a/routers/api/v1/repo/cloudbrain.go b/routers/api/v1/repo/cloudbrain.go
index 90746a8f37..ece216daf9 100755
--- a/routers/api/v1/repo/cloudbrain.go
+++ b/routers/api/v1/repo/cloudbrain.go
@@ -693,7 +693,7 @@ func CloudbrainDownloadLogFile(ctx *context.Context) {
 		}
 	}
 	if fileName != "" {
-		prefix := "/" + setting.CBCodePathPrefix + job.JobName + "/model"
+		prefix := "/" + setting.CBCodePathPrefix + job.JobName + logDir
 		filePath := setting.Attachment.Minio.RealPath + setting.Attachment.Minio.Bucket + prefix + "/" + fileName
 		// Read the file contents into a byte slice
 		data, err := ioutil.ReadFile(filePath)
diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go
index 3c270ca941..9330872426 100644
--- a/routers/repo/aisafety.go
+++ b/routers/repo/aisafety.go
@@ -313,7 +313,7 @@ func queryTaskStatusFromModelSafetyTestServer(job *models.Cloudbrain) {
 				log.Info("The task is running....")
 			} else {
 				job.EndTime = timeutil.TimeStampNow()
-				job.Duration = (job.EndTime.AsTime().Unix() - job.StartTime.AsTime().Unix()) / 1000
+				job.Duration = (job.EndTime.AsTime().Unix() - job.StartTime.AsTime().Unix())
 				job.TrainJobDuration = models.ConvertDurationToStr(job.Duration)
 				if result.Data.Code == 0 {
 					job.ResultJson = result.Data.StandardJson