Fixed another problem in rake task