Allow to use with Rails 5