Update template specialization to reflect API changes.