Fix image styling on update