diff --git a/src/MaaCore/Task/Infrast/InfrastTrainingTask.cpp b/src/MaaCore/Task/Infrast/InfrastTrainingTask.cpp
index f91b00b3f0..4facf38655 100644
--- a/src/MaaCore/Task/Infrast/InfrastTrainingTask.cpp
+++ b/src/MaaCore/Task/Infrast/InfrastTrainingTask.cpp
@@ -137,8 +137,7 @@ bool asst::InfrastTrainingTask::level_analyze(const cv::Mat& image)
 
 bool asst::InfrastTrainingTask::training_completed()
 {
-    ProcessTask task(*this, { "InfrastTrainingProcessing", "InfrastTrainingCompleted" });
-    return task.run() && task.get_last_task_name() == "InfrastTrainingCompleted";
+    return ProcessTask(*this, { "InfrastTrainingCompleted" }).run();
 }
 
 std::optional<std::string> asst::InfrastTrainingTask::time_left_analyze(const cv::Mat& image)