JKIT || Close Menu Canvas when menu on click

/**
 * Close Menu Canvas when menu on click
 */
function jkit_close_menu_canvas_when_menu_on_click() {
    ?>
    <script>
        /**
         * Close Menu Canvas when menu on click
         */
        (function ($) {
            $(document).on('ready', function () {
                const wrapper = $('.jeg-elementor-kit.jkit-nav-menu');
                const menuWrapper = wrapper.find('.jkit-menu-wrapper');
                const menu = menuWrapper.find('.jkit-menu-container .menu-item')

                menu.on('click', function () {
                    menuWrapper.removeClass('active');
                });
            });
        })(jQuery);
    </script>
    <?php
}
add_action( 'wp_head', 'jkit_close_menu_canvas_when_menu_on_click' );