Fix broken rake task