Support Rails 4 too