diff --git a/src/main/kotlin/dsl/ast/parser/ReturnParser.kt b/src/main/kotlin/dsl/ast/parser/ReturnParser.kt
index 12e7b80..b1f2d88 100644
--- a/src/main/kotlin/dsl/ast/parser/ReturnParser.kt
+++ b/src/main/kotlin/dsl/ast/parser/ReturnParser.kt
@@ -9,7 +9,7 @@ class ReturnParser : Parser() {
     override fun tryToParse(input: TokenList): ParserOutput {
         return allOf(
             terminal(TokenType.RETURN),
-            optional(SubexpressionParser())
+            optional(ExpressionParser())
         ) {
             ReturnNode(it[1])
         }.parse(input)