diff --git a/lib/rex/proto/ntlm.rb.ut.rb b/lib/rex/proto/ntlm.rb.ut.rb index 0ad2d4d0cc..a19e432d56 100644 --- a/lib/rex/proto/ntlm.rb.ut.rb +++ b/lib/rex/proto/ntlm.rb.ut.rb @@ -1,4 +1,7 @@ #!/usr/bin/env ruby + +$:.unshift(File.join(File.dirname(__FILE__), '..', '..')) + require 'test/unit' require 'rex/proto/ntlm' require 'rex/socket' diff --git a/lib/rex/proto/rfb.rb.ut.rb b/lib/rex/proto/rfb.rb.ut.rb index 29f39d0405..3333e61d04 100644 --- a/lib/rex/proto/rfb.rb.ut.rb +++ b/lib/rex/proto/rfb.rb.ut.rb @@ -1,5 +1,7 @@ #!/usr/bin/env ruby +$:.unshift(File.join(File.dirname(__FILE__), '..', '..')) + ## # $Id: $ ##