Fix the rake tasks