Fixes compiling zk itest module

Signed-off-by: Mario Krizmanic <mario.krizmanic@gmail.com>