fix compilation errors

Signed-off-by: Blaž Hrastnik <blaz.hrast@gmail.com>