fixed minor bug and update for new gem