more modification

Signed-off-by: fuqiang89 <513914894@qq.com>