diff --git a/src/Nodes/ArrayNode.php b/src/Nodes/ArrayNode.php
index 6de8d43..79bb2f4 100644
--- a/src/Nodes/ArrayNode.php
+++ b/src/Nodes/ArrayNode.php
@@ -27,9 +27,9 @@ public function __construct(string $name, ?CompoundNode $parent = null)
 
     /**
      * @param string|NodeType $type
-     * @return AbstractNode
+     * @return ScalarNode|ArrayNode|ObjectNode|FileNode
      */
-    public function each(string|NodeType $type): AbstractNode
+    public function each(NodeType|string $type): AbstractNode
     {
         $className = $this->resolveNodeClassName($type);
         $this->failIfTypeChanged($className);