update for new version of gem