Same as previous commit

diff --git a/lib/menu-walker.php b/lib/menu-walker.php
index 78775bb..a65fec9 100644 (file)
--- a/lib/menu-walker.php
+++ b/lib/menu-walker.php
@@ -56,7 +56,9 @@ class Glm_Theme_Hotel_Top_Bar_Walker extends Walker_Nav_Menu {
         $element->has_children = ! empty( $children_elements[ $element->ID ] );
         $element->classes[] = ( $element->current || $element->current_item_ancestor ) ? 'active' : '';
         $element->classes[] = ( $element->has_children && 1 !== $max_depth ) ? 'has-dropdown' : '';
-        $element->classes[] = ( count($children_elements[$element->ID]) > 4 ) ? 'submenu-two-columns' : '';
+        if ( array_key_exists($element->ID, $children_elements ) ) {
+            $element->classes[] = ( count($children_elements[$element->ID]) > 4 ) ? 'submenu-two-columns' : '';
+        }
         $element->classes[] = ( $element->post_parent == HOTEL_PAGE_ID && $mainHotelLevelCounter < 5 ) ? 'drop-right' : 'drop-left';
         parent::display_element( $element, $children_elements, $max_depth, $depth, $args, $output );
     }