Fix build for vault >= 0.3.0