Fix minor image bug