fix compilation problem

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