Rake task fix