Modified Rakefile to fix bug