Remove top level child if, add it to the two individual conditions

Author: VeraZab

src/components/containers/Fold.js

@@ -26,35 +26,34 @@ class Fold extends Component {
     this.foldVisible = false;
 
     React.Children.forEach(nextProps.children, child => {
-      if (child) {
-        if (child.props.attr) {
-          let plotProps;
-          if (child.type.supplyPlotProps) {
-            plotProps = child.type.supplyPlotProps(child.props, nextContext);
-            if (child.type.modifyPlotProps) {
-              child.type.modifyPlotProps(child.props, nextContext, plotProps);
-            }
-          } else {
-            plotProps = unpackPlotProps(child.props, nextContext);
-          }
-
-          if (plotProps.isVisible) {
-            this.foldVisible = true;
-            return;
+      if (child && child.props.attr) {
+        let plotProps;
+        if (child.type.supplyPlotProps) {
+          plotProps = child.type.supplyPlotProps(child.props, nextContext);
+          if (child.type.modifyPlotProps) {
+            child.type.modifyPlotProps(child.props, nextContext, plotProps);
           }
+        } else {
+          plotProps = unpackPlotProps(child.props, nextContext);
         }
 
-        // allow custom components in folds to automatically show up,
-        // except for Folds of Folds, which should keep their visibility rules
-        if (
-          !child.type.plotly_editor_traits ||
-          (child.type.plotly_editor_traits &&
-            !child.type.plotly_editor_traits.foldable)
-        ) {
+        if (plotProps.isVisible) {
           this.foldVisible = true;
           return;
         }
       }
+
+      // allow custom components in folds to automatically show up,
+      // except for Folds of Folds, which should keep their visibility rules
+      if (
+        child &&
+        (!child.type.plotly_editor_traits ||
+          (child.type.plotly_editor_traits &&
+            !child.type.plotly_editor_traits.foldable))
+      ) {
+        this.foldVisible = true;
+        return;
+      }
     });
   }