Updated for v2 API changes