From 790c8367714c2e9b17470bec78c52953ee95d037 Mon Sep 17 00:00:00 2001 From: Karol Stasiak Date: Fri, 11 Feb 2022 21:47:39 +0100 Subject: [PATCH] Fix handling of non-top-level return dispatch statements --- src/main/scala/millfork/node/Node.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/main/scala/millfork/node/Node.scala b/src/main/scala/millfork/node/Node.scala index e1a3f117..14d83f6d 100644 --- a/src/main/scala/millfork/node/Node.scala +++ b/src/main/scala/millfork/node/Node.scala @@ -722,7 +722,7 @@ case class StandardReturnDispatchLabel(labels:List[Expression]) extends ReturnDi } case class ReturnDispatchBranch(label: ReturnDispatchLabel, function: Expression, params: List[Expression]) extends Node { - def getAllExpressions: List[Expression] = label.getAllExpressions ++ params + def getAllExpressions: List[Expression] = label.getAllExpressions ++ params :+ function } case class ReturnDispatchStatement(indexer: Expression, params: List[LhsExpression], branches: List[ReturnDispatchBranch]) extends ExecutableStatement {