Avoid compile err

Signed-off-by: Keven Liu <airk908@gmail.com>