Update the vault classes to use the new API.