public override void OnStateEnter(Animator animator, AnimatorStateInfo stateInfo, int layerIndex)
    {
        BossBehaviour bossBehaviour = animator.transform.root.GetComponent <BossBehaviour>();

        if (bossBehaviour)
        {
            bossBehaviour.SetAllowMovement(true);
        }
    }