Fix handling of non-top-level return dispatch statements

This commit is contained in:
Karol Stasiak 2022-02-11 21:47:39 +01:00
parent 6af84d1628
commit 790c836771
1 changed files with 1 additions and 1 deletions

View File

@ -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 {