PRD-4379: Adjust code after API change.