Updated gem to work