Restored it to previous functionality in yeoman