Update for API changes
Signed-off-by:thehutch <th3hutch@yahoo.co.uk>