Update src/ast/Python3VisitorImpl.java

suggested by santo

Co-authored-by: Santo Cariotti <santo@dcariotti.me>

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);
     }
 
     /**