diff --git a/src/Tree/Entity/Repository/MaterializedPathRepository.php b/src/Tree/Entity/Repository/MaterializedPathRepository.php
index 1c03500762..291a767772 100644
--- a/src/Tree/Entity/Repository/MaterializedPathRepository.php
+++ b/src/Tree/Entity/Repository/MaterializedPathRepository.php
@@ -138,7 +138,7 @@ public function getPathQuery($node)
      *
      * @param object $node
      *
-     * @return array list of Nodes in path
+     * @return array<int, object> list of Nodes in path
      */
     public function getPath($node)
     {
diff --git a/src/Tree/Entity/Repository/NestedTreeRepository.php b/src/Tree/Entity/Repository/NestedTreeRepository.php
index 70553c9d4f..b2b9eaf7b8 100644
--- a/src/Tree/Entity/Repository/NestedTreeRepository.php
+++ b/src/Tree/Entity/Repository/NestedTreeRepository.php
@@ -403,7 +403,7 @@ public function childrenQuery($node = null, $direct = false, $sortByField = null
      * @param string|string[]      $direction   Sort order ('asc'|'desc'|'ASC'|'DESC'). If $sortByField is an array, this may also be an array with matching number of elements
      * @param bool                 $includeNode Flag indicating whether the given node should be included in the results
      *
-     * @return array<int, object> List of children or null on failure
+     * @return array<int, object> List of children
      *
      * @phpstan-param 'asc'|'desc'|'ASC'|'DESC'|array<int, 'asc'|'desc'|'ASC'|'DESC'> $direction
      */
@@ -425,7 +425,7 @@ public function getChildrenQuery($node = null, $direct = false, $sortByField = n
     }
 
     /**
-     * @return array<int, object|null>
+     * @return array<int, object>
      */
     public function getChildren($node = null, $direct = false, $sortByField = null, $direction = 'ASC', $includeNode = false)
     {
@@ -511,7 +511,7 @@ public function getLeafsQuery($root = null, $sortByField = null, $direction = 'A
      * @param string $sortByField field name to sort by
      * @param string $direction   sort direction : "ASC" or "DESC"
      *
-     * @return array
+     * @return array<int, object>
      *
      * @phpstan-param 'asc'|'desc'|'ASC'|'DESC' $direction
      */
@@ -593,7 +593,7 @@ public function getNextSiblingsQuery($node, $includeSelf = false)
      * @param object $node
      * @param bool   $includeSelf include the node itself
      *
-     * @return array
+     * @return array<int, object>
      */
     public function getNextSiblings($node, $includeSelf = false)
     {
@@ -672,7 +672,7 @@ public function getPrevSiblingsQuery($node, $includeSelf = false)
      * @param object $node
      * @param bool   $includeSelf include the node itself
      *
-     * @return array
+     * @return array<int, object>
      */
     public function getPrevSiblings($node, $includeSelf = false)
     {
diff --git a/src/Tree/RepositoryInterface.php b/src/Tree/RepositoryInterface.php
index 483f46e4ce..7bc95b5aca 100644
--- a/src/Tree/RepositoryInterface.php
+++ b/src/Tree/RepositoryInterface.php
@@ -37,7 +37,7 @@ public function getRootNodes($sortByField = null, $direction = 'asc');
      * @param array  $options     Options, see {@see RepositoryUtilsInterface::buildTree()} for supported keys
      * @param bool   $includeNode Flag indicating whether the given node should be included in the results
      *
-     * @return array
+     * @return array<int, object>
      */
     public function getNodesHierarchy($node = null, $direct = false, array $options = [], $includeNode = false);
 
@@ -50,7 +50,7 @@ public function getNodesHierarchy($node = null, $direct = false, array $options
      * @param string|string[]      $direction   Sort order ('asc'|'desc'|'ASC'|'DESC'). If $sortByField is an array, this may also be an array with matching number of elements
      * @param bool                 $includeNode Include the root node in results?
      *
-     * @return array|null List of children or null on failure
+     * @return array<int, object> List of children
      *
      * @phpstan-param 'asc'|'desc'|'ASC'|'DESC'|array<int, 'asc'|'desc'|'ASC'|'DESC'> $direction
      */