Set, show, wrap in code blocks

1 files changed, 5 insertions, 2 deletions

diff --git a/src/parse/mod.rs b/src/parse/mod.rs
index 86d18cba..4ef1c96f 100644
--- a/src/parse/mod.rs
+++ b/src/parse/mod.rs
@@ -246,7 +246,10 @@ fn markup_node(p: &mut Parser, at_start: &mut bool) {
         | NodeKind::While
         | NodeKind::For
         | NodeKind::Import
-        | NodeKind::Include => markup_expr(p),
+        | NodeKind::Include
+        | NodeKind::Break
+        | NodeKind::Continue
+        | NodeKind::Return => markup_expr(p),
 
         // Code and content block.
         NodeKind::LeftBrace => code_block(p),
@@ -965,7 +968,7 @@ fn continue_expr(p: &mut Parser) -> ParseResult {
 fn return_expr(p: &mut Parser) -> ParseResult {
     p.perform(NodeKind::ReturnExpr, |p| {
         p.assert(NodeKind::Return);
-        if !p.eof() {
+        if !p.at(NodeKind::Comma) && !p.eof() {
             expr(p)?;
         }
         Ok(())