Fix vaspio_set to work with new antiferromagnetic transformations