a little modification

Signed-off-by: Zhao Xianghu <xianghuzhao@gmail.com>