Changes to reflect changes in API

Signed-off-by: Alan Gardner <alan@armoin.com>