Tweak to Rakefile