Update for API changes

Signed-off-by: Chris Sanders <zidane@outlook.com>
Update for API changes

Signed-off-by: Chris Sanders <zidane@outlook.com>