improve code

Signed-off-by: chennanfei <chennanfei2006@163.com>