Fixup

zoonru · Jan 17, 2024 · 823c918 · 823c918
1 parent 4e5e306
commit 823c918
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/Psalm/Internal/PhpTraverser/CustomTraverser.php b/src/Psalm/Internal/PhpTraverser/CustomTraverser.php
@@ -29,7 +29,7 @@ public function __construct()
      * @param Node $node node to traverse
      * @return Node Result of traversal (may be original node or new one)
      */
-    protected function traverseNode(Node $node): void
+    protected function traverseNode(Node $node): Node
     {
         foreach ($node->getSubNodeNames() as $name) {
             $subNode = &$node->$name;
@@ -60,7 +60,7 @@ protected function traverseNode(Node $node): void
                 }
 
                 if ($traverseChildren) {
-                    $this->traverseNode($subNode);
+                    $subNode = $this->traverseNode($subNode);
                     if ($this->stopTraversal) {
                         break;
                     }
@@ -88,6 +88,8 @@ protected function traverseNode(Node $node): void
                 }
             }
         }
+
+        return $node;
     }
 
     /**