fix the compile issue

Signed-off-by: Le Min <lemin9538@gmail.com>