diff --git a/build.rb b/build.rb
index 3de6117..9557674 100755
--- a/build.rb
+++ b/build.rb
@@ -1,5 +1,7 @@
 #!/usr/bin/env ruby
 
+$LOAD_PATH << File.dirname(__FILE__)
+
 require 'compiler'
 require 'asm/text'
 require 'asm/binary'
diff --git a/test/test.rb b/test/test.rb
index e84f845..46c21bd 100755
--- a/test/test.rb
+++ b/test/test.rb
@@ -1,9 +1,6 @@
 #!/usr/bin/env ruby
 
-ROOT = Dir.pwd.sub(/\/test.*$/, '')
-$LOAD_PATH << ROOT
-
-require 'build'
+require '../build'
 
 # usage: test.rb <func> [outdir] [binformat] [format]