a bit modificataion

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