Fixed issue with Rake task