Rakefile in yard-0.5.8 vs Rakefile in yard-0.6.0

@@ -1,16 +1,18 @@
 require File.dirname(__FILE__) + '/lib/yard'
 require 'rbconfig'
 
+YARD::VERSION.replace(ENV['YARD_VERSION']) if ENV['YARD_VERSION']
 WINDOWS = (Config::CONFIG['host_os'] =~ /mingw|win32|cygwin/ ? true : false) rescue false
 SUDO = WINDOWS ? '' : 'sudo'
 
 task :default => :specs
 
 desc "Builds the gem"
 task :gem do
-  sh "gem build yard.gemspec"
+  load 'yard.gemspec'
+  Gem::Builder.new(SPEC).build
 end
 
 desc "Installs the gem"
 task :install => :gem do 
   sh "#{SUDO} gem install yard-#{YARD::VERSION}.gem --no-rdoc --no-ri"
@@ -40,7 +42,6 @@
   warn "warn: RSpec tests not available. `gem install rspec` to enable them."
 end
 
 YARD::Rake::YardocTask.new do |t|
   t.options += ['--title', "YARD #{YARD::VERSION} Documentation"]
-  t.after = lambda { `cp -R docs/images/ doc/images/` }
 end