Hm I guess we know if the image was added or removed when we create the imageDiff? Maybe we could just pass that information along in the args to the component