Fix compilation error

Signed-off-by: Alec Ari <neotheuser@ymail.com>