Improve getColumns for attributeVectors