def METHOD_NAME(self, unit):
    super().METHOD_NAME(unit)
    unit.set_unit_flag(UnitFlags.UNIT_FLAG_FLEEING, True)
    if not self.seek_assist or not unit.combat_target:
        return True
    # Should search assistance, search for a friendly unit.
    units = unit.get_map().get_surrounding_units_by_location(unit.location, unit.map_id, unit.instance_id,
                                                             FLEE_ASSISTANCE_RADIUS)[0].values()
    for unit in units:
        if not unit.threat_manager.unit_can_assist_help_call(unit, unit.combat_target):
            continue
        self.assist_unit = unit
        break
    return True