Distinguish pre-condition and post-condition loops

1 files changed, 2 insertions, 1 deletions

diff --git a/ast.c b/ast.c
index 3e25c89a55..2d219092c7 100644
--- a/ast.c
+++ b/ast.c
@@ -374,7 +374,8 @@ node_children(rb_ast_t *ast, NODE *node)
         goto loop;
       case NODE_UNTIL:
       loop:
-        return rb_ary_new_from_node_args(ast, 2, node->nd_cond, node->nd_body);
+        return rb_ary_push(rb_ary_new_from_node_args(ast, 2, node->nd_cond, node->nd_body),
+                           (node->nd_state ? Qtrue : Qfalse));
       case NODE_ITER:
       case NODE_FOR:
         return rb_ary_new_from_node_args(ast, 2, node->nd_iter, node->nd_body);