Fixing rake task.