Make tests compile after API change.