Fix build issue

Signed-off-by: Max Liao <maxliaops@gmail.com>