Update for API changes

Signed-off-by: Ollie <dzineit@live.co.uk>