Fix boilerplate to work with new framework.