Fixed build

Signed-off-by: Saggi Mizrahi <ficoos@gmail.com>