Minor update to vit