some code modify

Signed-off-by: isongfei <isongfei@gmail.com>