DOC: Adapt calls in example to changed signature from 1626e0f