Updates according API changes

Signed-off-by: Ladislav Klenovic <klenovic@nerdrobot.net>