Can that be safely done in the init function there? I'm happy to put up a PR for that.