Small fix

Signed-off-by: Harry Chen <373685963@qq.com>