Update code ref wrapper to always make images