[fix] Updated Tests to reflect the new API
Merge pull request #22 from wallner/OSNG-233

[fix] Updated Tests to reflect the new API