you could just return here from the beginning

return $node
    ->children()
    ...
    ->end()
; or no? i wonder if that is allowed with the fluent interface