diff --git a/petr-parse/src/parse_to_ast.rs b/petr-parse/src/parse_to_ast.rs index 36efc2e..73c38d8 100644 --- a/petr-parse/src/parse_to_ast.rs +++ b/petr-parse/src/parse_to_ast.rs @@ -165,7 +165,7 @@ impl Parse for FunctionDeclaration { p.token(Token::CloseParen)?; seq }; - p.token(Token::ReturnsKeyword)?; + p.one_of([Token::ReturnsKeyword, Token::ReturnsSymbol])?; let return_type = p.parse()?; let body = p.parse()?; Some(Self {