fix flash after edit