Fix up the gem for Rails 4