fix update_with_media

This turned out to be easier than I thought it'd be. `update_with_media` is fairly well tested so I'm not sure additional tests are necessary as these changes are below the surface. I'm happy to add some if you think it's needed.