small fix

Signed-off-by: latelee <latelee@163.com>