Reference: fix edit image problem