Fixed build

Signed-off-by: Olli Vanhoja <olli.vanhoja@cs.helsinki.fi>