diff --git a/src/test/scala/leon/test/LeonTestSuiteWithProgram.scala b/src/test/scala/leon/test/LeonTestSuiteWithProgram.scala
index 17d65364de16d2a37b26be4a6ca147ac5c5ccb41..75987ea886097cad2cb304f9edda160014dc7da0 100644
--- a/src/test/scala/leon/test/LeonTestSuiteWithProgram.scala
+++ b/src/test/scala/leon/test/LeonTestSuiteWithProgram.scala
@@ -4,7 +4,7 @@ package leon.test
 
 import leon._
 import leon.purescala.Definitions.Program
-import leon.LeonContext
+import leon.purescala.FunctionClosure
 import leon.utils._
 import leon.frontends.scalac.ExtractionPhase
 
@@ -23,7 +23,8 @@ trait LeonTestSuiteWithProgram extends fixture.FunSuite {
   val pipeline =
     TemporaryInputPhase andThen
     ExtractionPhase andThen
-    PreprocessingPhase
+    PreprocessingPhase andThen
+    new FunctionClosure
 
   val sources: List[String]