Fix rake task