Small change to rake task