Updated to reflect api changes

Signed-off-by: Tobias Sjondin <tsjondin@op5.com>