updated tests inline with vault changes