Updated for new API

Signed-off-by: Nader Hendawi <nader.hendawi@axial.net>