fix TreeUtil.findParent()

Author: brachy84

src/main/java/com/cleanroommc/modularui/utils/TreeUtil.java

@@ -3,7 +3,6 @@
 import com.cleanroommc.modularui.ModularUI;
 import com.cleanroommc.modularui.api.ITreeNode;
 import com.cleanroommc.modularui.api.widget.IWidget;
-import com.cleanroommc.modularui.screen.ModularPanel;
 import com.cleanroommc.modularui.widget.sizer.ResizeNode;
 
 import com.google.common.collect.AbstractIterator;
@@ -289,6 +288,7 @@ public static <T extends ITreeNode<T>, R extends ITreeNode<T>> List<R> flatListB
      * @param test   test which the widget has to pass
      * @return the first matching widget
      */
+    @Nullable
     public static <T extends ITreeNode<T>> T findFirst(T parent, @NotNull Predicate<T> test) {
         return foreachChildWithResult(parent, w -> {
             if (test.test(w)) {
@@ -306,6 +306,7 @@ public static <T extends ITreeNode<T>> T findFirst(T parent, @NotNull Predicate<
      * @param test   test which the widget has to pass
      * @return the first matching widget
      */
+    @Nullable
     @SuppressWarnings("unchecked")
     public static <T extends ITreeNode<T>, R extends ITreeNode<T>> R findFirst(T parent, Class<R> type, @Nullable Predicate<R> test) {
         return foreachChildWithResult(parent, t -> {
@@ -318,15 +319,15 @@ public static <T extends ITreeNode<T>, R extends ITreeNode<T>> R findFirst(T par
         }, true);
     }
 
+    @Nullable
     public static <T extends ITreeNode<T>> T findParent(T parent, Predicate<T> filter) {
-        if (parent == null) return null;
-        while (!(parent instanceof ModularPanel)) {
+        while (parent != null) {
             if (filter.test(parent)) {
                 return parent;
             }
             parent = parent.getParent();
         }
-        return filter.test(parent) ? parent : null;
+        return null;
     }
 
     @Nullable