diff options
Diffstat (limited to 'src/ast')
| -rw-r--r-- | src/ast/Python3VisitorImpl.java | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/ast/Python3VisitorImpl.java b/src/ast/Python3VisitorImpl.java index c5f1bfa..bef75cf 100644 --- a/src/ast/Python3VisitorImpl.java +++ b/src/ast/Python3VisitorImpl.java @@ -536,11 +536,11 @@ public class Python3VisitorImpl extends Python3ParserBaseVisitor<Node> { exprlist.add(visit(c)); } Comp_forContext cfc = ctx.comp_for(); + Node comp = null; if (cfc != null) { - Node comp = visit(ctx.comp_for()); - return new TestlistCompNode(exprlist, comp); + comp = visit(ctx.comp_for()); } - return new TestlistCompNode(exprlist, null); + return new TestlistCompNode(exprlist, comp); } /** |