diff --git a/src/main/scala/leon/UnitElimination.scala b/src/main/scala/leon/UnitElimination.scala
index 6d3aa84562cbe56fee5b018e9b93578dd0b9b6ce..3d56d537723f5286c950d50bdc84b755d3715180 100644
--- a/src/main/scala/leon/UnitElimination.scala
+++ b/src/main/scala/leon/UnitElimination.scala
@@ -3,6 +3,7 @@ package leon
 import purescala.Common._
 import purescala.Definitions._
 import purescala.Trees._
+import xlang.Trees._
 import purescala.Extractors._
 import purescala.TypeTrees._