Adding compatability for Rails 4.1