Update to reflect latest Aurora API changes