Fix the script to bundle as well