Make it work with Rails 3 and 4