Stopping Enemy y rotation when player is jump

How i could make enemy attack to stop Y rotation When player is Jump?

Enemy Attack Script

    Transform player;

    override public void OnStateEnter(Animator animator, AnimatorStateInfo stateInfo, int layerIndex)
    {
        player = GameObject.FindGameObjectWithTag("Player").transform;
    }
    override public void OnStateUpdate(Animator animator, AnimatorStateInfo stateInfo, int layerIndex)
    {
        animator.transform.LookAt(player);
        float distance = Vector3.Distance(player.position, animator.transform.position);
        if (distance > 9)
            animator.SetBool("Attack",false);
    }

my enemy end up like this

Screenshot 2023-10-29 142411

Ohh i resolve this ^^

Vector3 playerPosition = new Vector3(player.position.x, animator.transform.position.y, player.position.z);
animator.transform.LookAt(playerPosition);