Allow gem to work with Rails 4.1