Minor mods to rake task.