code updated

Signed-off-by: jiarongxu <roger@fangohr.com>