rake task fix