Fix in rake task