$element->classes[] = ( $element->current || $element->current_item_ancestor ) ? 'active' : '';
$element->classes[] = ( $element->has_children && 1 !== $max_depth ) ? 'has-dropdown' : '';
$element->classes[] = ( $element->post_parent == 0 && ($mainLevelCounter < 6 || $mainLevelCounter == 1 )) ? '' : 'drop-left';
- $element->classes[] = ( $element->post_parent == 0 && $mainLevelCounter < 6 && $mainLevelCounter !== 6 ) ? '' : $mainLevelCounter;
+ $element->classes[] = ( $element->post_parent == 0 && ($mainLevelCounter < 6 || $mainLevelCounter == 1 )) ? '' : $mainLevelCounter;
parent::display_element( $element, $children_elements, $max_depth, $depth, $args, $output );
}