diff --git a/lua/app/module/battle/team/battle_team.lua b/lua/app/module/battle/team/battle_team.lua
index 5b9ebbe3..3afe8c6c 100644
--- a/lua/app/module/battle/team/battle_team.lua
+++ b/lua/app/module/battle/team/battle_team.lua
@@ -58,8 +58,8 @@ function BattleTeam:useSkill(matchType, callback)
 	unit:useSkill(1, callback)
 end
 
-function BattleTeam:mainUnitUseAllSkills()
-	self.mainUnit:useAllSkills()
+function BattleTeam:mainUnitUseAllSkills(callback)
+	self.mainUnit:useAllSkills(callback)
 end
 
 function BattleTeam:tick(dt)