Fix compiling error

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