Fixed AfterBattleUpdate