return is implicit void

1 files changed, 7 insertions, 1 deletions

diff --git a/src/compiler.rs b/src/compiler.rs
index 4c59d86..bdc5901 100644
--- a/src/compiler.rs
+++ b/src/compiler.rs
@@ -1752,7 +1752,13 @@ impl Compiler {
 
             (Token::Ret, ..) => {
                 self.eat();
-                self.expression(block);
+                if self.peek() == Token::Newline {
+                    self.eat();
+                    let nil = self.add_constant(Value::Nil);
+                    add_op(self, block, Op::Constant(nil));
+                } else {
+                    self.expression(block);
+                }
                 add_op(self, block, Op::Return);
             }