[Bug] Generating images error

Thanks! Should be fixed now. I think it would have occurred in cases where the image gen iframe was removed just after the image finished generating.