Allow for Rails 4