Update patch_bake.py