diff --git a/ArknightsUID/arknightsuid_ap/draw_ap_card.py b/ArknightsUID/arknightsuid_ap/draw_ap_card.py
index b7d7b01..8ebbfc3 100644
--- a/ArknightsUID/arknightsuid_ap/draw_ap_card.py
+++ b/ArknightsUID/arknightsuid_ap/draw_ap_card.py
@@ -275,7 +275,7 @@ async def draw_ap_img(uid: str) -> Image.Image:
         img.paste(grey_bar_bg1_img, (-20, 910), grey_bar_bg1_img)
 
     # training char check
-    if player_info.building.training:
+    if player_info.building.training and player_info.building.training.trainee:
         training_char = player_info.building.training.trainee.charId
         remain_secs = player_info.building.training.remainSecs
         remain_time = 0
diff --git a/ArknightsUID/utils/models/skland/models.py b/ArknightsUID/utils/models/skland/models.py
index b931d02..5305f4d 100644
--- a/ArknightsUID/utils/models/skland/models.py
+++ b/ArknightsUID/utils/models/skland/models.py
@@ -410,13 +410,13 @@ class BuildingLabor(Struct):
 class BuildingTraining(Struct):
     slotId: str
     level: int
-    trainee: BuildingTrainingTrainee
     remainPoint: float
     speed: float
     lastUpdateTime: int
     remainSecs: int
     slotState: int
-    trainer: BuildingTrainingTrainer | None = None
+    trainee: BuildingTrainingTrainee | None 
+    trainer: BuildingTrainingTrainer | None
 
 
 class BuildingHire(Struct):