Fixes imageSize enhancement to save to reference attribute.