diff --git a/lua/app/module/battle/skill/battle_rogue_skill_handle.lua b/lua/app/module/battle/skill/battle_rogue_skill_handle.lua
index 6398c233..fc4d29c5 100644
--- a/lua/app/module/battle/skill/battle_rogue_skill_handle.lua
+++ b/lua/app/module/battle/skill/battle_rogue_skill_handle.lua
@@ -235,10 +235,10 @@ local _addSkillEffect = function(skillId, skillInfo, battleBaseData, battleContr
             return
         end
         local buffEntity = BattleBuffEntity:create()
-        buffEntity:init(effect, actionTeam:getUnitComp().unitEntity)
+        buffEntity:init(effect, actionTeam:getMainUnit().unitEntity)
         buffEntity:setTargetSide(skillInfo.obj)
 
-        actionTeam:getUnitComp():takeEffect(buffEntity, target)
+        actionTeam:getMainUnit():takeEffect(buffEntity, target)
     end
 end