Update for new API

Signed-off-by: Michael Barton <mail@michaelbarton.me.uk>