Update for new yeoman gen