The download button downloads the original image instead of the AI modified image.

I recently tried using Adobe Firefly's generative AI feature. It's excellent at removing the background; however, when I click the download button after removing the background from the image, it downloads the original image, which still has the background. 

The download button should download the new AI generated/modifed image instead of the original uploaded image.