diff --git a/lua/app/module/battle/controller/battle_controller.lua b/lua/app/module/battle/controller/battle_controller.lua
index 697b76a8..42389623 100644
--- a/lua/app/module/battle/controller/battle_controller.lua
+++ b/lua/app/module/battle/controller/battle_controller.lua
@@ -70,6 +70,7 @@ function BattleController:findNextDefUnit()
 end
 
 function BattleController:onDefDead()
+    self.defTeam:getMainUnit():playDead()
 end
 
 function BattleController:tick(dt)
diff --git a/lua/app/module/battle/controller/battle_controller_stage.lua b/lua/app/module/battle/controller/battle_controller_stage.lua
index 6a099be6..f68481d9 100644
--- a/lua/app/module/battle/controller/battle_controller_stage.lua
+++ b/lua/app/module/battle/controller/battle_controller_stage.lua
@@ -154,10 +154,6 @@ function BattleControllerStage:findNextDefUnit()
     self:_stageGenerateNextMonster()
 end
 
-function BattleControllerStage:onDefDead()
-    self.defTeam:getMainUnit():playDead()
-end
-
 function BattleControllerStage:controllBattleEnd()
     self.combatReport = {
         battleType = GConst.BattleConst.BATTLE_TYPE.STAGE,