Rails 4 compatibility

Great rake task, thank you very much :-)