update for new yo generators