Updating gem to be compatible with Rails4