Ticket #21534: 21534.diff

wp-includes/class-wp-walker.php

@@ -123 +123 @@
                 $id_field = $this->db_fields['id'];
 
                 //display this element
-                if ( isset( $args[0] ) && is_array( $args[0] ) )
-                        $args[0]['has_children'] = ! empty( $children_elements[$element->$id_field] );
+                if ( isset( $args[0] ) ) {
+                        if ( is_array( $args[0] ) )
+                                $args[0]['has_children'] = ! empty( $children_elements[$element->$id_field] );
+                        elseif ( is_object( $args[0] ) )
+                                $args[0]->has_children = ! empty( $children_elements[$element->$id_field] );
+                }
+
                 $cb_args = array_merge( array(&$output, $element, $depth), $args);
                 call_user_func_array(array($this, 'start_el'), $cb_args);