Update gem to work on travis