Fixes for v2 API changes