diff --git a/src/main/scala/leon/xlang/ImperativeCodeElimination.scala b/src/main/scala/leon/xlang/ImperativeCodeElimination.scala
index 364c6fbb2f823e833cb82675f7fcc5ca1383ed1a..7a00cb7f8d128a6b9d495950d5cd0bc444093c7c 100644
--- a/src/main/scala/leon/xlang/ImperativeCodeElimination.scala
+++ b/src/main/scala/leon/xlang/ImperativeCodeElimination.scala
@@ -316,7 +316,7 @@ object ImperativeCodeElimination extends UnitPhase[Program] {
                       (v.toVariable, TupleSelect(newRes.toVariable, i+2)): (Expr, Expr)}.toMap + 
                     (res.toVariable -> TupleSelect(newRes.toVariable, 1)),
                   postBody)
-                Lambda(Seq(newRes), newBody)
+                Lambda(Seq(newRes), newBody).setPos(post)
               })
 
               val (bodyRes, bodyScope, bodyFun) = toFunction(b)(state.withFunDef(fd, newFd, modifiedVars))