1 files changed, 7 insertions, 3 deletions
diff --git a/rakelib/stage.rake b/rakelib/stage.rake
index 80f704d..6f61cae 100644
--- a/rakelib/stage.rake
+++ b/rakelib/stage.rake
@@ -21,16 +21,20 @@ def assemble_stage(output, compiler, source)
   end
 end
 
+library = []
+
 Dir.glob('boot/*.s').each do |assembly_source|
-  target_object = Pathname.new('build/boot') + Pathname.new(assembly_source).basename.sub_ext('.o')
+  source_basename = Pathname.new(assembly_source).basename
+  target_object = Pathname.new('build/boot') + source_basename.sub_ext('.o')
 
   file target_object.to_s => [assembly_source, 'build/boot'] do |t|
     sh CROSS_GCC, '-c', '-o', t.name, assembly_source
   end
+  library << assembly_source unless source_basename.to_s.start_with? 'stage'
 end
 
 desc 'Initial stage'
-file 'build/boot/stage1' => ['build/boot/tokenizer.o', 'build/boot/stage1.o', 'build/boot/common-boot.o'] do |t|
+file 'build/boot/stage1' => ['build/boot/stage1.o', *library] do |t|
   sh CROSS_GCC, '-nostdlib', '-o', t.name, *t.prerequisites
 end
 
@@ -43,7 +47,7 @@ file 'build/boot/stage2a.s' => ['build/boot/stage1', 'boot/stage2.elna'] do |t|
 end
 
 ['build/boot/stage2a', 'build/boot/stage2b'].each do |exe|
-  file exe => [exe.ext('.s'), 'build/boot/common-boot.o'] do |t|
+  file exe => [exe.ext('.s'), *library] do |t|
     sh CROSS_GCC, '-nostdlib', '-o', t.name, *t.prerequisites
   end
 end