diff --git a/core-java-8/src/main/java/com/baeldung/threadpool/TreeNode.java b/core-java-8/src/main/java/com/baeldung/threadpool/TreeNode.java
index 72b1f756a8..9b43152074 100644
--- a/core-java-8/src/main/java/com/baeldung/threadpool/TreeNode.java
+++ b/core-java-8/src/main/java/com/baeldung/threadpool/TreeNode.java
@@ -1,9 +1,9 @@
 package com.baeldung.threadpool;
 
-import java.util.Arrays;
-import java.util.HashSet;
 import java.util.Set;
 
+import com.google.common.collect.Sets;
+
 public class TreeNode {
 
     int value;
@@ -12,8 +12,7 @@ public class TreeNode {
 
     public TreeNode(int value, TreeNode... children) {
         this.value = value;
-        this.children = new HashSet<>();
-        this.children.addAll(Arrays.asList(children));
+        this.children = Sets.newHashSet(children);
     }
 
 }