public override IAIState GetNextState(IAIState currentState)
        {
            WaspAIState beeState = (WaspAIState)currentState;

            switch (beeState.action)
            {
            case WaspAIAction.Move:
                beeState.action = WaspAIAction.Attack;
                break;

            case WaspAIAction.None:
            case WaspAIAction.Attack:
                beeState.action = WaspAIAction.Move;
                break;
            }
            return(beeState);
        }
        /******* Monobehavior Methods *******/

        public void Awake()
        {
            _wasp    = GetComponent <Wasp>();
            _aiState = new WaspAIState();
            StartAI();
        }