1 files changed, 4 insertions, 2 deletions
diff --git a/src/codegen.c b/src/codegen.c
index ebc9b32de..ca2269d85 100644
--- a/src/codegen.c
+++ b/src/codegen.c
@@ -1556,8 +1556,10 @@ codegen(codegen_scope *s, node *tree, int val)
       genop(s, MKOP_sBx(OP_JMP, s->loop->pc1 - s->pc));
     }
     else {
-      codegen(s, tree, VAL);
-      pop();
+      if (tree) {
+        codegen(s, tree, VAL);
+        pop();
+      }
       genop_peep(s, MKOP_AB(OP_RETURN, cursp(), OP_R_NORMAL), NOVAL);
     }
     if (val) push();