1 files changed, 5 insertions, 6 deletions
diff --git a/stage2/Stage2.scala b/stage2/Stage2.scala
index eaf776e..9eb47fe 100644
--- a/stage2/Stage2.scala
+++ b/stage2/Stage2.scala
@@ -3,9 +3,7 @@ import java.io._
 import java.util.{Set=>_,_}
 
 object Stage2 extends Stage2Base{
-  def getBuild(context: Context) = {
-    new Lib( context.logger ).loadRoot( context ).finalBuild( context.cwd )
-  }
+  def getBuild(context: Context): Dependency = DirectoryDependency( context, None )
 
   def run( args: Stage2Args ): ExitCode = {
     import args.logger
@@ -34,9 +32,10 @@ object Stage2 extends Stage2Base{
       null,
       args.loop
     )
-    val first = lib.loadRoot( context )
-    val build = first.finalBuild( context.cwd )
-    val code = lib.callReflective(build, task, context)
+    val code = lib.callReflective(
+      DirectoryDependency( context, None ),
+      task, context
+    )
     logger.stage2(s"Stage2 end with exit code "+code.integer)
     code
   }