Fixed another bug in rake task