diff --git a/src/test/scala/leon/regression/repair/RepairSuite.scala b/src/test/scala/leon/regression/repair/RepairSuite.scala
index 254c6a2f5e468fc5af3b22d6b9ae4b2530f05fc9..3724b017044b57aa80a74cbd84d092da3e9b0804 100644
--- a/src/test/scala/leon/regression/repair/RepairSuite.scala
+++ b/src/test/scala/leon/regression/repair/RepairSuite.scala
@@ -23,15 +23,17 @@ class RepairSuite extends LeonRegressionSuite {
   )
   
   for (file <- filesInResourceDir("regression/repair/", _.endsWith(".scala"))) {
-    if (!(fileToFun contains file.getName)) {
-      fail(s"Don't know which function to repair for ${file.getName}")
-    }
+
     val path = file.getAbsoluteFile.toString
     val name = file.getName
 
-    val ctx = createLeonContext("--parallel", "--timeout=180", "--solvers=smt-z3")
-
     test(name) {
+      if (!(fileToFun contains file.getName)) {
+        fail(s"Don't know which function to repair for ${file.getName}")
+      }
+
+      val ctx = createLeonContext("--parallel", "--timeout=180", "--solvers=smt-z3", s"--functions=${fileToFun(name)}")
+
       pipeline.run(ctx, List(path))
       if(ctx.reporter.errorCount > 0) {
         fail("Errors during repair:\n"+ctx.reporter.asInstanceOf[TestSilentReporter].lastErrors.mkString("\n"))