Fix to rakefile