Add decorator back to build process