Fix for rake task