Adapt the code to the interface change in r196108.