Fixed to use the new rake task style