Fix save/remove media from attribute