diff --git a/models/cloudbrain.go b/models/cloudbrain.go
index 4ee0b7b33..11368f076 100755
--- a/models/cloudbrain.go
+++ b/models/cloudbrain.go
@@ -1395,7 +1395,7 @@ func UpdateTrainJobVersion(job *Cloudbrain) error {
 func updateJobTrainVersion(e Engine, job *Cloudbrain) error {
 	var sess *xorm.Session
 	sess = e.Where("job_id = ? AND version_name=?", job.JobID, job.VersionName)
-	_, err := sess.Cols("status", "train_job_duration").Update(job)
+	_, err := sess.Cols("status", "train_job_duration", "start_time", "end_time").Update(job)
 	return err
 }