diff --git a/src/compiler.rs b/src/compiler.rs
index 4f9e3d2..f185437 100644
--- a/src/compiler.rs
+++ b/src/compiler.rs
@@ -701,6 +701,7 @@ impl StmtVisitor for Compiler<'_> {
             let nil = self.insert_constant(self.create_nil())?;
             self.emit(stmt_line_number(stmt), Op::PushConstant(nil));
         }
+        self.emit(stmt_line_number(stmt), Op::Return);
         Ok(())
     }
     fn visit_if_stmt(&mut self, stmt: &IfStmt) -> Result<()> {