Fixed build errors

Signed-off-by: Jack Yen <jack@jackyen.com>