Rake Task Fix