Fix build for beta2 API changes.