Add compatibility for Rails 4.0