Fix for #168 to reflect the Core API change
Merge pull request #218 from Rise-Vision/chore/issue-168-api-change

Fix for #168 to reflect the Core API change