small fix

Signed-off-by: Pengcheng Zhu <zhupc123@gmail.com>