Made minor change to the rake task