diff --git a/src/main/scala/millfork/node/Node.scala b/src/main/scala/millfork/node/Node.scala index 51d1367f..9ec3e996 100644 --- a/src/main/scala/millfork/node/Node.scala +++ b/src/main/scala/millfork/node/Node.scala @@ -163,7 +163,7 @@ object ForDirection extends Enumeration { } case class ForStatement(variable: String, start: Expression, end: Expression, direction: ForDirection.Value, body: List[ExecutableStatement]) extends ExecutableStatement { - override def getAllExpressions: List[Expression] = start :: end :: body.flatMap(_.getAllExpressions) + override def getAllExpressions: List[Expression] = VariableExpression(variable) :: start :: end :: body.flatMap(_.getAllExpressions) } case class DoWhileStatement(body: List[ExecutableStatement], condition: Expression) extends ExecutableStatement {