Rakefile in genspec-0.2.1 vs Rakefile in genspec-0.2.2

@@ -5,14 +5,19 @@
   raise "tests failed" unless system *args
 end
 
 task :default do
   run "rspec", "spec"
-  ENV['RAILS'] = '1'
+  ENV['USE_RAILS'] = '1'
   run "rspec", "spec"
 end
 
-require 'rake/rdoctask'
+begin
+  require "rdoc/task"
+rescue LoadError
+  require 'rake/rdoctask'
+end
+
 Rake::RDocTask.new do |rdoc|
   version = GenSpec::VERSION
 
   rdoc.rdoc_dir = 'rdoc'
   rdoc.title = "genspec #{version}"