Fix no image model being returned.