diff --git a/src/ORM/Tree/TreeTrait.php b/src/ORM/Tree/TreeTrait.php
index a623c996..e10a3465 100644
--- a/src/ORM/Tree/TreeTrait.php
+++ b/src/ORM/Tree/TreeTrait.php
@@ -30,7 +30,7 @@ public function getRootNodes(string $rootAlias = 't'): array
     /**
      * Returns a node hydrated with its children and parents
      *
-     * @return TreeNodeInterface[]|ArrayAccess|null
+     * @return TreeNodeInterface|ArrayAccess|null
      */
     public function getTree(string $path = '', string $rootAlias = 't', array $extraParams = [])
     {
@@ -53,7 +53,7 @@ public function getTreeExceptNodeAndItsChildrenQB(
     /**
      * Extracts the root node and constructs a tree using flat resultset
      *
-     * @return ArrayAccess|TreeNodeInterface[]|null
+     * @return ArrayAccess|TreeNodeInterface|null
      */
     public function buildTree(array $results)
     {