Update theme JS to work with new build process.