diff --git a/api/internal/logic/ai/getcentertasklistlogic.go b/api/internal/logic/ai/getcentertasklistlogic.go
index bbae384b..b7fa034f 100644
--- a/api/internal/logic/ai/getcentertasklistlogic.go
+++ b/api/internal/logic/ai/getcentertasklistlogic.go
@@ -113,6 +113,10 @@ func (l *GetCenterTaskListLogic) updateAiTaskStatus(mu *sync.RWMutex, ch chan<-
 					wg.Done()
 					return
 				}
+				if trainingTask == nil {
+					wg.Done()
+					return
+				}
 				t.Status = trainingTask.Status
 				t.StartTime = trainingTask.Start
 				t.EndTime = trainingTask.End
diff --git a/api/internal/logic/core/pagelisttasklogic.go b/api/internal/logic/core/pagelisttasklogic.go
index f1858a4a..db5ea094 100644
--- a/api/internal/logic/core/pagelisttasklogic.go
+++ b/api/internal/logic/core/pagelisttasklogic.go
@@ -255,6 +255,10 @@ func (l *PageListTaskLogic) updateAiTaskStatus(tasks []*types.TaskModel, ch chan
 				wg.Done()
 				return
 			}
+			if trainingTask == nil {
+				wg.Done()
+				return
+			}
 			t.Status = trainingTask.Status
 			t.StartTime = trainingTask.Start
 			t.EndTime = trainingTask.End