a bit modify

Signed-off-by: TomWu <phperwuhan@gmail.com>