Fixups for new API.