Fix the warnings from the previous commit

1 files changed, 4 insertions, 4 deletions

diff --git a/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java b/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java
index 6e21f27..ddef800 100644
--- a/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java
+++ b/src/main/java/au/id/zancanaro/javacheck/ShrinkTree.java
@@ -56,17 +56,17 @@ public class ShrinkTree<T> {
     public static <T> Stream<ShrinkTree<T>[]> promoteChildren(ShrinkTree<? super T>[] trees) {
         return IntStream.range(0, trees.length)
                 .mapToObj(index -> trees[index].getChildren().map(child ->
-                        IntStream.range(0, trees.length)
+                        (ShrinkTree<T>[]) IntStream.range(0, trees.length)
                                 .mapToObj(i -> (i == index ? child : trees[i]))
                                 .toArray(ShrinkTree[]::new)))
-                .flatMap(x -> x)
-                .map(x -> (ShrinkTree<T>[]) x);
+                .flatMap(x -> x);
     }
 
+    @SuppressWarnings("unchecked")
     public static <T> Stream<ShrinkTree<T>[]> removeChildren(ShrinkTree<? super T>[] trees) {
         return IntStream.range(0, trees.length)
                 .mapToObj(index ->
-                        IntStream.range(0, trees.length)
+                        (ShrinkTree<T>[]) IntStream.range(0, trees.length)
                                 .filter(i -> i != index)
                                 .mapToObj(i -> trees[i])
                                 .toArray(ShrinkTree[]::new));