Как в данном коде сделать так, чтобы персонаж не мог двигаться в сторону во время прыжка (находясь в воздухе)

Я попытался сделать проверку всего движения (через if), есть ли персонаж на земле, или нет, проблема частично решилась, но при заходе на лестницу, он застряёт и больше не двигается, увеличение параметра groundDistance ничего не дало. Заранее спасибо! Код взят отсюда https://youtu.be/_QajrabyTJc

using UnityEngine;

   public class PlayerMovementV2 : MonoBehaviour
   {
    public float speed = 12f;
    public float gravity = -9.81f;
    public CharacterController controller;

    public float jumpHeight = 3f;

    public Transform groundCheck;
    public float groundDistance = 0.4f;
    public LayerMask groundMask;

    Vector3 velocity;
    bool isGrounded;



    void Update()
    {
        isGrounded = Physics.CheckSphere(groundCheck.position, groundDistance, groundMask);
        if (isGrounded && velocity.y < 0)
        {
            velocity.y = -2f;
        }

        //add movement logic
        float x = Input.GetAxis("Horizontal");
        float z = Input.GetAxis("Vertical");

        Vector3 move = transform.right * x + transform.forward * z;

        controller.Move(move * speed * Time.deltaTime);

        //add jump logic
        if (Input.GetButtonDown ("Jump") && isGrounded)
        {
            velocity.y = Mathf.Sqrt(jumpHeight * -2f * gravity);
        }

        //add gravity logic
        velocity.y += gravity * Time.deltaTime;
        controller.Move(velocity * Time.deltaTime);



    }
}