a bit modification.

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