Move setup to 'build' hook