Add support to Rails 4