fixed gem to work with ruby 2