Fix spec tests.

1 files changed, 10 insertions, 0 deletions

diff --git a/Rakefile b/Rakefile
index 7053a89..a2728e4 100644
--- a/Rakefile
+++ b/Rakefile
@@ -1,6 +1,16 @@
 require "bundler/gem_tasks"
 require "rspec/core/rake_task"
 
+# Add the include path for the fit4ruby library. We assume it is located in
+# the same directory as the postrunner directory.
+fit4ruby = File.realpath(File.join(File.dirname(__FILE__), '..',
+                                                'fit4ruby', 'lib'))
+if ENV['RUBYLIB']
+  ENV['RUBYLIB'] += ":#{fit4ruby}"
+else
+  ENV['RUBYLIB'] = fit4ruby
+end
+
 RSpec::Core::RakeTask.new
 
 task :default => :spec