small change

Signed-off-by: Levi Wang <wangli5665@gmail.com>