Updated gem to work with Rails 4