Fix import in Rakefile