Fix broken rake tasks